1 //===--- SemaExprObjC.cpp - Semantic Analysis for ObjC Expressions --------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 //  This file implements semantic analysis for Objective-C expressions.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Sema/SemaInternal.h"
15 #include "clang/Sema/Lookup.h"
16 #include "clang/Sema/Scope.h"
17 #include "clang/Sema/ScopeInfo.h"
18 #include "clang/Sema/Initialization.h"
19 #include "clang/Analysis/DomainSpecific/CocoaConventions.h"
20 #include "clang/AST/ASTContext.h"
21 #include "clang/AST/DeclObjC.h"
22 #include "clang/AST/ExprObjC.h"
23 #include "clang/AST/StmtVisitor.h"
24 #include "clang/AST/TypeLoc.h"
25 #include "llvm/ADT/SmallString.h"
26 #include "clang/Lex/Preprocessor.h"
27 
28 using namespace clang;
29 using namespace sema;
30 using llvm::makeArrayRef;
31 
32 ExprResult Sema::ParseObjCStringLiteral(SourceLocation *AtLocs,
33                                         Expr **strings,
34                                         unsigned NumStrings) {
35   StringLiteral **Strings = reinterpret_cast<StringLiteral**>(strings);
36 
37   // Most ObjC strings are formed out of a single piece.  However, we *can*
38   // have strings formed out of multiple @ strings with multiple pptokens in
39   // each one, e.g. @"foo" "bar" @"baz" "qux"   which need to be turned into one
40   // StringLiteral for ObjCStringLiteral to hold onto.
41   StringLiteral *S = Strings[0];
42 
43   // If we have a multi-part string, merge it all together.
44   if (NumStrings != 1) {
45     // Concatenate objc strings.
46     llvm::SmallString<128> StrBuf;
47     SmallVector<SourceLocation, 8> StrLocs;
48 
49     for (unsigned i = 0; i != NumStrings; ++i) {
50       S = Strings[i];
51 
52       // ObjC strings can't be wide or UTF.
53       if (!S->isAscii()) {
54         Diag(S->getLocStart(), diag::err_cfstring_literal_not_string_constant)
55           << S->getSourceRange();
56         return true;
57       }
58 
59       // Append the string.
60       StrBuf += S->getString();
61 
62       // Get the locations of the string tokens.
63       StrLocs.append(S->tokloc_begin(), S->tokloc_end());
64     }
65 
66     // Create the aggregate string with the appropriate content and location
67     // information.
68     S = StringLiteral::Create(Context, StrBuf,
69                               StringLiteral::Ascii, /*Pascal=*/false,
70                               Context.getPointerType(Context.CharTy),
71                               &StrLocs[0], StrLocs.size());
72   }
73 
74   // Verify that this composite string is acceptable for ObjC strings.
75   if (CheckObjCString(S))
76     return true;
77 
78   // Initialize the constant string interface lazily. This assumes
79   // the NSString interface is seen in this translation unit. Note: We
80   // don't use NSConstantString, since the runtime team considers this
81   // interface private (even though it appears in the header files).
82   QualType Ty = Context.getObjCConstantStringInterface();
83   if (!Ty.isNull()) {
84     Ty = Context.getObjCObjectPointerType(Ty);
85   } else if (getLangOptions().NoConstantCFStrings) {
86     IdentifierInfo *NSIdent=0;
87     std::string StringClass(getLangOptions().ObjCConstantStringClass);
88 
89     if (StringClass.empty())
90       NSIdent = &Context.Idents.get("NSConstantString");
91     else
92       NSIdent = &Context.Idents.get(StringClass);
93 
94     NamedDecl *IF = LookupSingleName(TUScope, NSIdent, AtLocs[0],
95                                      LookupOrdinaryName);
96     if (ObjCInterfaceDecl *StrIF = dyn_cast_or_null<ObjCInterfaceDecl>(IF)) {
97       Context.setObjCConstantStringInterface(StrIF);
98       Ty = Context.getObjCConstantStringInterface();
99       Ty = Context.getObjCObjectPointerType(Ty);
100     } else {
101       // If there is no NSConstantString interface defined then treat this
102       // as error and recover from it.
103       Diag(S->getLocStart(), diag::err_no_nsconstant_string_class) << NSIdent
104         << S->getSourceRange();
105       Ty = Context.getObjCIdType();
106     }
107   } else {
108     IdentifierInfo *NSIdent = &Context.Idents.get("NSString");
109     NamedDecl *IF = LookupSingleName(TUScope, NSIdent, AtLocs[0],
110                                      LookupOrdinaryName);
111     if (ObjCInterfaceDecl *StrIF = dyn_cast_or_null<ObjCInterfaceDecl>(IF)) {
112       Context.setObjCConstantStringInterface(StrIF);
113       Ty = Context.getObjCConstantStringInterface();
114       Ty = Context.getObjCObjectPointerType(Ty);
115     } else {
116       // If there is no NSString interface defined then treat constant
117       // strings as untyped objects and let the runtime figure it out later.
118       Ty = Context.getObjCIdType();
119     }
120   }
121 
122   return new (Context) ObjCStringLiteral(S, Ty, AtLocs[0]);
123 }
124 
125 ExprResult Sema::BuildObjCEncodeExpression(SourceLocation AtLoc,
126                                       TypeSourceInfo *EncodedTypeInfo,
127                                       SourceLocation RParenLoc) {
128   QualType EncodedType = EncodedTypeInfo->getType();
129   QualType StrTy;
130   if (EncodedType->isDependentType())
131     StrTy = Context.DependentTy;
132   else {
133     if (!EncodedType->getAsArrayTypeUnsafe() && //// Incomplete array is handled.
134         !EncodedType->isVoidType()) // void is handled too.
135       if (RequireCompleteType(AtLoc, EncodedType,
136                          PDiag(diag::err_incomplete_type_objc_at_encode)
137                              << EncodedTypeInfo->getTypeLoc().getSourceRange()))
138         return ExprError();
139 
140     std::string Str;
141     Context.getObjCEncodingForType(EncodedType, Str);
142 
143     // The type of @encode is the same as the type of the corresponding string,
144     // which is an array type.
145     StrTy = Context.CharTy;
146     // A C++ string literal has a const-qualified element type (C++ 2.13.4p1).
147     if (getLangOptions().CPlusPlus || getLangOptions().ConstStrings)
148       StrTy.addConst();
149     StrTy = Context.getConstantArrayType(StrTy, llvm::APInt(32, Str.size()+1),
150                                          ArrayType::Normal, 0);
151   }
152 
153   return new (Context) ObjCEncodeExpr(StrTy, EncodedTypeInfo, AtLoc, RParenLoc);
154 }
155 
156 ExprResult Sema::ParseObjCEncodeExpression(SourceLocation AtLoc,
157                                            SourceLocation EncodeLoc,
158                                            SourceLocation LParenLoc,
159                                            ParsedType ty,
160                                            SourceLocation RParenLoc) {
161   // FIXME: Preserve type source info ?
162   TypeSourceInfo *TInfo;
163   QualType EncodedType = GetTypeFromParser(ty, &TInfo);
164   if (!TInfo)
165     TInfo = Context.getTrivialTypeSourceInfo(EncodedType,
166                                              PP.getLocForEndOfToken(LParenLoc));
167 
168   return BuildObjCEncodeExpression(AtLoc, TInfo, RParenLoc);
169 }
170 
171 ExprResult Sema::ParseObjCSelectorExpression(Selector Sel,
172                                              SourceLocation AtLoc,
173                                              SourceLocation SelLoc,
174                                              SourceLocation LParenLoc,
175                                              SourceLocation RParenLoc) {
176   ObjCMethodDecl *Method = LookupInstanceMethodInGlobalPool(Sel,
177                              SourceRange(LParenLoc, RParenLoc), false, false);
178   if (!Method)
179     Method = LookupFactoryMethodInGlobalPool(Sel,
180                                           SourceRange(LParenLoc, RParenLoc));
181   if (!Method)
182     Diag(SelLoc, diag::warn_undeclared_selector) << Sel;
183 
184   if (!Method ||
185       Method->getImplementationControl() != ObjCMethodDecl::Optional) {
186     llvm::DenseMap<Selector, SourceLocation>::iterator Pos
187       = ReferencedSelectors.find(Sel);
188     if (Pos == ReferencedSelectors.end())
189       ReferencedSelectors.insert(std::make_pair(Sel, SelLoc));
190   }
191 
192   // In ARC, forbid the user from using @selector for
193   // retain/release/autorelease/dealloc/retainCount.
194   if (getLangOptions().ObjCAutoRefCount) {
195     switch (Sel.getMethodFamily()) {
196     case OMF_retain:
197     case OMF_release:
198     case OMF_autorelease:
199     case OMF_retainCount:
200     case OMF_dealloc:
201       Diag(AtLoc, diag::err_arc_illegal_selector) <<
202         Sel << SourceRange(LParenLoc, RParenLoc);
203       break;
204 
205     case OMF_None:
206     case OMF_alloc:
207     case OMF_copy:
208     case OMF_finalize:
209     case OMF_init:
210     case OMF_mutableCopy:
211     case OMF_new:
212     case OMF_self:
213     case OMF_performSelector:
214       break;
215     }
216   }
217   QualType Ty = Context.getObjCSelType();
218   return new (Context) ObjCSelectorExpr(Ty, Sel, AtLoc, RParenLoc);
219 }
220 
221 ExprResult Sema::ParseObjCProtocolExpression(IdentifierInfo *ProtocolId,
222                                              SourceLocation AtLoc,
223                                              SourceLocation ProtoLoc,
224                                              SourceLocation LParenLoc,
225                                              SourceLocation RParenLoc) {
226   ObjCProtocolDecl* PDecl = LookupProtocol(ProtocolId, ProtoLoc);
227   if (!PDecl) {
228     Diag(ProtoLoc, diag::err_undeclared_protocol) << ProtocolId;
229     return true;
230   }
231 
232   QualType Ty = Context.getObjCProtoType();
233   if (Ty.isNull())
234     return true;
235   Ty = Context.getObjCObjectPointerType(Ty);
236   return new (Context) ObjCProtocolExpr(Ty, PDecl, AtLoc, RParenLoc);
237 }
238 
239 /// Try to capture an implicit reference to 'self'.
240 ObjCMethodDecl *Sema::tryCaptureObjCSelf() {
241   // Ignore block scopes: we can capture through them.
242   DeclContext *DC = CurContext;
243   while (true) {
244     if (isa<BlockDecl>(DC)) DC = cast<BlockDecl>(DC)->getDeclContext();
245     else if (isa<EnumDecl>(DC)) DC = cast<EnumDecl>(DC)->getDeclContext();
246     else break;
247   }
248 
249   // If we're not in an ObjC method, error out.  Note that, unlike the
250   // C++ case, we don't require an instance method --- class methods
251   // still have a 'self', and we really do still need to capture it!
252   ObjCMethodDecl *method = dyn_cast<ObjCMethodDecl>(DC);
253   if (!method)
254     return 0;
255 
256   ImplicitParamDecl *self = method->getSelfDecl();
257   assert(self && "capturing 'self' in non-definition?");
258 
259   // Mark that we're closing on 'this' in all the block scopes, if applicable.
260   for (unsigned idx = FunctionScopes.size() - 1;
261        isa<BlockScopeInfo>(FunctionScopes[idx]);
262        --idx) {
263     BlockScopeInfo *blockScope = cast<BlockScopeInfo>(FunctionScopes[idx]);
264     unsigned &captureIndex = blockScope->CaptureMap[self];
265     if (captureIndex) break;
266 
267     bool nested = isa<BlockScopeInfo>(FunctionScopes[idx-1]);
268     blockScope->Captures.push_back(
269               BlockDecl::Capture(self, /*byref*/ false, nested, /*copy*/ 0));
270     captureIndex = blockScope->Captures.size(); // +1
271   }
272 
273   return method;
274 }
275 
276 static QualType stripObjCInstanceType(ASTContext &Context, QualType T) {
277   if (T == Context.getObjCInstanceType())
278     return Context.getObjCIdType();
279 
280   return T;
281 }
282 
283 QualType Sema::getMessageSendResultType(QualType ReceiverType,
284                                         ObjCMethodDecl *Method,
285                                     bool isClassMessage, bool isSuperMessage) {
286   assert(Method && "Must have a method");
287   if (!Method->hasRelatedResultType())
288     return Method->getSendResultType();
289 
290   // If a method has a related return type:
291   //   - if the method found is an instance method, but the message send
292   //     was a class message send, T is the declared return type of the method
293   //     found
294   if (Method->isInstanceMethod() && isClassMessage)
295     return stripObjCInstanceType(Context, Method->getSendResultType());
296 
297   //   - if the receiver is super, T is a pointer to the class of the
298   //     enclosing method definition
299   if (isSuperMessage) {
300     if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
301       if (ObjCInterfaceDecl *Class = CurMethod->getClassInterface())
302         return Context.getObjCObjectPointerType(
303                                         Context.getObjCInterfaceType(Class));
304   }
305 
306   //   - if the receiver is the name of a class U, T is a pointer to U
307   if (ReceiverType->getAs<ObjCInterfaceType>() ||
308       ReceiverType->isObjCQualifiedInterfaceType())
309     return Context.getObjCObjectPointerType(ReceiverType);
310   //   - if the receiver is of type Class or qualified Class type,
311   //     T is the declared return type of the method.
312   if (ReceiverType->isObjCClassType() ||
313       ReceiverType->isObjCQualifiedClassType())
314     return stripObjCInstanceType(Context, Method->getSendResultType());
315 
316   //   - if the receiver is id, qualified id, Class, or qualified Class, T
317   //     is the receiver type, otherwise
318   //   - T is the type of the receiver expression.
319   return ReceiverType;
320 }
321 
322 void Sema::EmitRelatedResultTypeNote(const Expr *E) {
323   E = E->IgnoreParenImpCasts();
324   const ObjCMessageExpr *MsgSend = dyn_cast<ObjCMessageExpr>(E);
325   if (!MsgSend)
326     return;
327 
328   const ObjCMethodDecl *Method = MsgSend->getMethodDecl();
329   if (!Method)
330     return;
331 
332   if (!Method->hasRelatedResultType())
333     return;
334 
335   if (Context.hasSameUnqualifiedType(Method->getResultType()
336                                                         .getNonReferenceType(),
337                                      MsgSend->getType()))
338     return;
339 
340   if (!Context.hasSameUnqualifiedType(Method->getResultType(),
341                                       Context.getObjCInstanceType()))
342     return;
343 
344   Diag(Method->getLocation(), diag::note_related_result_type_inferred)
345     << Method->isInstanceMethod() << Method->getSelector()
346     << MsgSend->getType();
347 }
348 
349 bool Sema::CheckMessageArgumentTypes(QualType ReceiverType,
350                                      Expr **Args, unsigned NumArgs,
351                                      Selector Sel, ObjCMethodDecl *Method,
352                                      bool isClassMessage, bool isSuperMessage,
353                                      SourceLocation lbrac, SourceLocation rbrac,
354                                      QualType &ReturnType, ExprValueKind &VK) {
355   if (!Method) {
356     // Apply default argument promotion as for (C99 6.5.2.2p6).
357     for (unsigned i = 0; i != NumArgs; i++) {
358       if (Args[i]->isTypeDependent())
359         continue;
360 
361       ExprResult Result = DefaultArgumentPromotion(Args[i]);
362       if (Result.isInvalid())
363         return true;
364       Args[i] = Result.take();
365     }
366 
367     unsigned DiagID;
368     if (getLangOptions().ObjCAutoRefCount)
369       DiagID = diag::err_arc_method_not_found;
370     else
371       DiagID = isClassMessage ? diag::warn_class_method_not_found
372                               : diag::warn_inst_method_not_found;
373     if (!getLangOptions().DebuggerSupport)
374       Diag(lbrac, DiagID)
375         << Sel << isClassMessage << SourceRange(lbrac, rbrac);
376 
377     // In debuggers, we want to use __unknown_anytype for these
378     // results so that clients can cast them.
379     if (getLangOptions().DebuggerSupport) {
380       ReturnType = Context.UnknownAnyTy;
381     } else {
382       ReturnType = Context.getObjCIdType();
383     }
384     VK = VK_RValue;
385     return false;
386   }
387 
388   ReturnType = getMessageSendResultType(ReceiverType, Method, isClassMessage,
389                                         isSuperMessage);
390   VK = Expr::getValueKindForType(Method->getResultType());
391 
392   unsigned NumNamedArgs = Sel.getNumArgs();
393   // Method might have more arguments than selector indicates. This is due
394   // to addition of c-style arguments in method.
395   if (Method->param_size() > Sel.getNumArgs())
396     NumNamedArgs = Method->param_size();
397   // FIXME. This need be cleaned up.
398   if (NumArgs < NumNamedArgs) {
399     Diag(lbrac, diag::err_typecheck_call_too_few_args)
400       << 2 << NumNamedArgs << NumArgs;
401     return false;
402   }
403 
404   bool IsError = false;
405   for (unsigned i = 0; i < NumNamedArgs; i++) {
406     // We can't do any type-checking on a type-dependent argument.
407     if (Args[i]->isTypeDependent())
408       continue;
409 
410     Expr *argExpr = Args[i];
411 
412     ParmVarDecl *param = Method->param_begin()[i];
413     assert(argExpr && "CheckMessageArgumentTypes(): missing expression");
414 
415     // Strip the unbridged-cast placeholder expression off unless it's
416     // a consumed argument.
417     if (argExpr->hasPlaceholderType(BuiltinType::ARCUnbridgedCast) &&
418         !param->hasAttr<CFConsumedAttr>())
419       argExpr = stripARCUnbridgedCast(argExpr);
420 
421     if (RequireCompleteType(argExpr->getSourceRange().getBegin(),
422                             param->getType(),
423                             PDiag(diag::err_call_incomplete_argument)
424                               << argExpr->getSourceRange()))
425       return true;
426 
427     InitializedEntity Entity = InitializedEntity::InitializeParameter(Context,
428                                                                       param);
429     ExprResult ArgE = PerformCopyInitialization(Entity, lbrac, Owned(argExpr));
430     if (ArgE.isInvalid())
431       IsError = true;
432     else
433       Args[i] = ArgE.takeAs<Expr>();
434   }
435 
436   // Promote additional arguments to variadic methods.
437   if (Method->isVariadic()) {
438     for (unsigned i = NumNamedArgs; i < NumArgs; ++i) {
439       if (Args[i]->isTypeDependent())
440         continue;
441 
442       ExprResult Arg = DefaultVariadicArgumentPromotion(Args[i], VariadicMethod, 0);
443       IsError |= Arg.isInvalid();
444       Args[i] = Arg.take();
445     }
446   } else {
447     // Check for extra arguments to non-variadic methods.
448     if (NumArgs != NumNamedArgs) {
449       Diag(Args[NumNamedArgs]->getLocStart(),
450            diag::err_typecheck_call_too_many_args)
451         << 2 /*method*/ << NumNamedArgs << NumArgs
452         << Method->getSourceRange()
453         << SourceRange(Args[NumNamedArgs]->getLocStart(),
454                        Args[NumArgs-1]->getLocEnd());
455     }
456   }
457   // diagnose nonnull arguments.
458   for (specific_attr_iterator<NonNullAttr>
459        i = Method->specific_attr_begin<NonNullAttr>(),
460        e = Method->specific_attr_end<NonNullAttr>(); i != e; ++i) {
461     CheckNonNullArguments(*i, Args, lbrac);
462   }
463 
464   DiagnoseSentinelCalls(Method, lbrac, Args, NumArgs);
465   return IsError;
466 }
467 
468 bool Sema::isSelfExpr(Expr *receiver) {
469   // 'self' is objc 'self' in an objc method only.
470   ObjCMethodDecl *method =
471     dyn_cast<ObjCMethodDecl>(CurContext->getNonClosureAncestor());
472   if (!method) return false;
473 
474   receiver = receiver->IgnoreParenLValueCasts();
475   if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(receiver))
476     if (DRE->getDecl() == method->getSelfDecl())
477       return true;
478   return false;
479 }
480 
481 // Helper method for ActOnClassMethod/ActOnInstanceMethod.
482 // Will search "local" class/category implementations for a method decl.
483 // If failed, then we search in class's root for an instance method.
484 // Returns 0 if no method is found.
485 ObjCMethodDecl *Sema::LookupPrivateClassMethod(Selector Sel,
486                                           ObjCInterfaceDecl *ClassDecl) {
487   ObjCMethodDecl *Method = 0;
488   // lookup in class and all superclasses
489   while (ClassDecl && !Method) {
490     if (ObjCImplementationDecl *ImpDecl = ClassDecl->getImplementation())
491       Method = ImpDecl->getClassMethod(Sel);
492 
493     // Look through local category implementations associated with the class.
494     if (!Method)
495       Method = ClassDecl->getCategoryClassMethod(Sel);
496 
497     // Before we give up, check if the selector is an instance method.
498     // But only in the root. This matches gcc's behaviour and what the
499     // runtime expects.
500     if (!Method && !ClassDecl->getSuperClass()) {
501       Method = ClassDecl->lookupInstanceMethod(Sel);
502       // Look through local category implementations associated
503       // with the root class.
504       if (!Method)
505         Method = LookupPrivateInstanceMethod(Sel, ClassDecl);
506     }
507 
508     ClassDecl = ClassDecl->getSuperClass();
509   }
510   return Method;
511 }
512 
513 ObjCMethodDecl *Sema::LookupPrivateInstanceMethod(Selector Sel,
514                                               ObjCInterfaceDecl *ClassDecl) {
515   ObjCMethodDecl *Method = 0;
516   while (ClassDecl && !Method) {
517     // If we have implementations in scope, check "private" methods.
518     if (ObjCImplementationDecl *ImpDecl = ClassDecl->getImplementation())
519       Method = ImpDecl->getInstanceMethod(Sel);
520 
521     // Look through local category implementations associated with the class.
522     if (!Method)
523       Method = ClassDecl->getCategoryInstanceMethod(Sel);
524     ClassDecl = ClassDecl->getSuperClass();
525   }
526   return Method;
527 }
528 
529 /// LookupMethodInType - Look up a method in an ObjCObjectType.
530 ObjCMethodDecl *Sema::LookupMethodInObjectType(Selector sel, QualType type,
531                                                bool isInstance) {
532   const ObjCObjectType *objType = type->castAs<ObjCObjectType>();
533   if (ObjCInterfaceDecl *iface = objType->getInterface()) {
534     // Look it up in the main interface (and categories, etc.)
535     if (ObjCMethodDecl *method = iface->lookupMethod(sel, isInstance))
536       return method;
537 
538     // Okay, look for "private" methods declared in any
539     // @implementations we've seen.
540     if (isInstance) {
541       if (ObjCMethodDecl *method = LookupPrivateInstanceMethod(sel, iface))
542         return method;
543     } else {
544       if (ObjCMethodDecl *method = LookupPrivateClassMethod(sel, iface))
545         return method;
546     }
547   }
548 
549   // Check qualifiers.
550   for (ObjCObjectType::qual_iterator
551          i = objType->qual_begin(), e = objType->qual_end(); i != e; ++i)
552     if (ObjCMethodDecl *method = (*i)->lookupMethod(sel, isInstance))
553       return method;
554 
555   return 0;
556 }
557 
558 /// LookupMethodInQualifiedType - Lookups up a method in protocol qualifier
559 /// list of a qualified objective pointer type.
560 ObjCMethodDecl *Sema::LookupMethodInQualifiedType(Selector Sel,
561                                               const ObjCObjectPointerType *OPT,
562                                               bool Instance)
563 {
564   ObjCMethodDecl *MD = 0;
565   for (ObjCObjectPointerType::qual_iterator I = OPT->qual_begin(),
566        E = OPT->qual_end(); I != E; ++I) {
567     ObjCProtocolDecl *PROTO = (*I);
568     if ((MD = PROTO->lookupMethod(Sel, Instance))) {
569       return MD;
570     }
571   }
572   return 0;
573 }
574 
575 /// HandleExprPropertyRefExpr - Handle foo.bar where foo is a pointer to an
576 /// objective C interface.  This is a property reference expression.
577 ExprResult Sema::
578 HandleExprPropertyRefExpr(const ObjCObjectPointerType *OPT,
579                           Expr *BaseExpr, SourceLocation OpLoc,
580                           DeclarationName MemberName,
581                           SourceLocation MemberLoc,
582                           SourceLocation SuperLoc, QualType SuperType,
583                           bool Super) {
584   const ObjCInterfaceType *IFaceT = OPT->getInterfaceType();
585   ObjCInterfaceDecl *IFace = IFaceT->getDecl();
586 
587   if (MemberName.getNameKind() != DeclarationName::Identifier) {
588     Diag(MemberLoc, diag::err_invalid_property_name)
589       << MemberName << QualType(OPT, 0);
590     return ExprError();
591   }
592 
593   IdentifierInfo *Member = MemberName.getAsIdentifierInfo();
594   SourceRange BaseRange = Super? SourceRange(SuperLoc)
595                                : BaseExpr->getSourceRange();
596   if (RequireCompleteType(MemberLoc, OPT->getPointeeType(),
597                           PDiag(diag::err_property_not_found_forward_class)
598                             << MemberName << BaseRange))
599     return ExprError();
600 
601   // Search for a declared property first.
602   if (ObjCPropertyDecl *PD = IFace->FindPropertyDeclaration(Member)) {
603     // Check whether we can reference this property.
604     if (DiagnoseUseOfDecl(PD, MemberLoc))
605       return ExprError();
606 
607     if (Super)
608       return Owned(new (Context) ObjCPropertyRefExpr(PD, Context.PseudoObjectTy,
609                                                      VK_LValue, OK_ObjCProperty,
610                                                      MemberLoc,
611                                                      SuperLoc, SuperType));
612     else
613       return Owned(new (Context) ObjCPropertyRefExpr(PD, Context.PseudoObjectTy,
614                                                      VK_LValue, OK_ObjCProperty,
615                                                      MemberLoc, BaseExpr));
616   }
617   // Check protocols on qualified interfaces.
618   for (ObjCObjectPointerType::qual_iterator I = OPT->qual_begin(),
619        E = OPT->qual_end(); I != E; ++I)
620     if (ObjCPropertyDecl *PD = (*I)->FindPropertyDeclaration(Member)) {
621       // Check whether we can reference this property.
622       if (DiagnoseUseOfDecl(PD, MemberLoc))
623         return ExprError();
624 
625       if (Super)
626         return Owned(new (Context) ObjCPropertyRefExpr(PD,
627                                                        Context.PseudoObjectTy,
628                                                        VK_LValue,
629                                                        OK_ObjCProperty,
630                                                        MemberLoc,
631                                                        SuperLoc, SuperType));
632       else
633         return Owned(new (Context) ObjCPropertyRefExpr(PD,
634                                                        Context.PseudoObjectTy,
635                                                        VK_LValue,
636                                                        OK_ObjCProperty,
637                                                        MemberLoc,
638                                                        BaseExpr));
639     }
640   // If that failed, look for an "implicit" property by seeing if the nullary
641   // selector is implemented.
642 
643   // FIXME: The logic for looking up nullary and unary selectors should be
644   // shared with the code in ActOnInstanceMessage.
645 
646   Selector Sel = PP.getSelectorTable().getNullarySelector(Member);
647   ObjCMethodDecl *Getter = IFace->lookupInstanceMethod(Sel);
648 
649   // May be founf in property's qualified list.
650   if (!Getter)
651     Getter = LookupMethodInQualifiedType(Sel, OPT, true);
652 
653   // If this reference is in an @implementation, check for 'private' methods.
654   if (!Getter)
655     Getter = IFace->lookupPrivateMethod(Sel);
656 
657   // Look through local category implementations associated with the class.
658   if (!Getter)
659     Getter = IFace->getCategoryInstanceMethod(Sel);
660   if (Getter) {
661     // Check if we can reference this property.
662     if (DiagnoseUseOfDecl(Getter, MemberLoc))
663       return ExprError();
664   }
665   // If we found a getter then this may be a valid dot-reference, we
666   // will look for the matching setter, in case it is needed.
667   Selector SetterSel =
668     SelectorTable::constructSetterName(PP.getIdentifierTable(),
669                                        PP.getSelectorTable(), Member);
670   ObjCMethodDecl *Setter = IFace->lookupInstanceMethod(SetterSel);
671 
672   // May be founf in property's qualified list.
673   if (!Setter)
674     Setter = LookupMethodInQualifiedType(SetterSel, OPT, true);
675 
676   if (!Setter) {
677     // If this reference is in an @implementation, also check for 'private'
678     // methods.
679     Setter = IFace->lookupPrivateMethod(SetterSel);
680   }
681   // Look through local category implementations associated with the class.
682   if (!Setter)
683     Setter = IFace->getCategoryInstanceMethod(SetterSel);
684 
685   if (Setter && DiagnoseUseOfDecl(Setter, MemberLoc))
686     return ExprError();
687 
688   if (Getter || Setter) {
689     if (Super)
690       return Owned(new (Context) ObjCPropertyRefExpr(Getter, Setter,
691                                                      Context.PseudoObjectTy,
692                                                      VK_LValue, OK_ObjCProperty,
693                                                      MemberLoc,
694                                                      SuperLoc, SuperType));
695     else
696       return Owned(new (Context) ObjCPropertyRefExpr(Getter, Setter,
697                                                      Context.PseudoObjectTy,
698                                                      VK_LValue, OK_ObjCProperty,
699                                                      MemberLoc, BaseExpr));
700 
701   }
702 
703   // Attempt to correct for typos in property names.
704   TypoCorrection Corrected = CorrectTypo(
705       DeclarationNameInfo(MemberName, MemberLoc), LookupOrdinaryName, NULL,
706       NULL, IFace, false, CTC_NoKeywords, OPT);
707   if (ObjCPropertyDecl *Property =
708       Corrected.getCorrectionDeclAs<ObjCPropertyDecl>()) {
709     DeclarationName TypoResult = Corrected.getCorrection();
710     Diag(MemberLoc, diag::err_property_not_found_suggest)
711       << MemberName << QualType(OPT, 0) << TypoResult
712       << FixItHint::CreateReplacement(MemberLoc, TypoResult.getAsString());
713     Diag(Property->getLocation(), diag::note_previous_decl)
714       << Property->getDeclName();
715     return HandleExprPropertyRefExpr(OPT, BaseExpr, OpLoc,
716                                      TypoResult, MemberLoc,
717                                      SuperLoc, SuperType, Super);
718   }
719   ObjCInterfaceDecl *ClassDeclared;
720   if (ObjCIvarDecl *Ivar =
721       IFace->lookupInstanceVariable(Member, ClassDeclared)) {
722     QualType T = Ivar->getType();
723     if (const ObjCObjectPointerType * OBJPT =
724         T->getAsObjCInterfacePointerType()) {
725       if (RequireCompleteType(MemberLoc, OBJPT->getPointeeType(),
726                               PDiag(diag::err_property_not_as_forward_class)
727                                 << MemberName << BaseExpr->getSourceRange()))
728         return ExprError();
729     }
730     Diag(MemberLoc,
731          diag::err_ivar_access_using_property_syntax_suggest)
732     << MemberName << QualType(OPT, 0) << Ivar->getDeclName()
733     << FixItHint::CreateReplacement(OpLoc, "->");
734     return ExprError();
735   }
736 
737   Diag(MemberLoc, diag::err_property_not_found)
738     << MemberName << QualType(OPT, 0);
739   if (Setter)
740     Diag(Setter->getLocation(), diag::note_getter_unavailable)
741           << MemberName << BaseExpr->getSourceRange();
742   return ExprError();
743 }
744 
745 
746 
747 ExprResult Sema::
748 ActOnClassPropertyRefExpr(IdentifierInfo &receiverName,
749                           IdentifierInfo &propertyName,
750                           SourceLocation receiverNameLoc,
751                           SourceLocation propertyNameLoc) {
752 
753   IdentifierInfo *receiverNamePtr = &receiverName;
754   ObjCInterfaceDecl *IFace = getObjCInterfaceDecl(receiverNamePtr,
755                                                   receiverNameLoc);
756 
757   bool IsSuper = false;
758   if (IFace == 0) {
759     // If the "receiver" is 'super' in a method, handle it as an expression-like
760     // property reference.
761     if (receiverNamePtr->isStr("super")) {
762       IsSuper = true;
763 
764       if (ObjCMethodDecl *CurMethod = tryCaptureObjCSelf()) {
765         if (CurMethod->isInstanceMethod()) {
766           QualType T =
767             Context.getObjCInterfaceType(CurMethod->getClassInterface());
768           T = Context.getObjCObjectPointerType(T);
769 
770           return HandleExprPropertyRefExpr(T->getAsObjCInterfacePointerType(),
771                                            /*BaseExpr*/0,
772                                            SourceLocation()/*OpLoc*/,
773                                            &propertyName,
774                                            propertyNameLoc,
775                                            receiverNameLoc, T, true);
776         }
777 
778         // Otherwise, if this is a class method, try dispatching to our
779         // superclass.
780         IFace = CurMethod->getClassInterface()->getSuperClass();
781       }
782     }
783 
784     if (IFace == 0) {
785       Diag(receiverNameLoc, diag::err_expected_ident_or_lparen);
786       return ExprError();
787     }
788   }
789 
790   // Search for a declared property first.
791   Selector Sel = PP.getSelectorTable().getNullarySelector(&propertyName);
792   ObjCMethodDecl *Getter = IFace->lookupClassMethod(Sel);
793 
794   // If this reference is in an @implementation, check for 'private' methods.
795   if (!Getter)
796     if (ObjCMethodDecl *CurMeth = getCurMethodDecl())
797       if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface())
798         if (ObjCImplementationDecl *ImpDecl = ClassDecl->getImplementation())
799           Getter = ImpDecl->getClassMethod(Sel);
800 
801   if (Getter) {
802     // FIXME: refactor/share with ActOnMemberReference().
803     // Check if we can reference this property.
804     if (DiagnoseUseOfDecl(Getter, propertyNameLoc))
805       return ExprError();
806   }
807 
808   // Look for the matching setter, in case it is needed.
809   Selector SetterSel =
810     SelectorTable::constructSetterName(PP.getIdentifierTable(),
811                                        PP.getSelectorTable(), &propertyName);
812 
813   ObjCMethodDecl *Setter = IFace->lookupClassMethod(SetterSel);
814   if (!Setter) {
815     // If this reference is in an @implementation, also check for 'private'
816     // methods.
817     if (ObjCMethodDecl *CurMeth = getCurMethodDecl())
818       if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface())
819         if (ObjCImplementationDecl *ImpDecl = ClassDecl->getImplementation())
820           Setter = ImpDecl->getClassMethod(SetterSel);
821   }
822   // Look through local category implementations associated with the class.
823   if (!Setter)
824     Setter = IFace->getCategoryClassMethod(SetterSel);
825 
826   if (Setter && DiagnoseUseOfDecl(Setter, propertyNameLoc))
827     return ExprError();
828 
829   if (Getter || Setter) {
830     if (IsSuper)
831     return Owned(new (Context) ObjCPropertyRefExpr(Getter, Setter,
832                                                    Context.PseudoObjectTy,
833                                                    VK_LValue, OK_ObjCProperty,
834                                                    propertyNameLoc,
835                                                    receiverNameLoc,
836                                           Context.getObjCInterfaceType(IFace)));
837 
838     return Owned(new (Context) ObjCPropertyRefExpr(Getter, Setter,
839                                                    Context.PseudoObjectTy,
840                                                    VK_LValue, OK_ObjCProperty,
841                                                    propertyNameLoc,
842                                                    receiverNameLoc, IFace));
843   }
844   return ExprError(Diag(propertyNameLoc, diag::err_property_not_found)
845                      << &propertyName << Context.getObjCInterfaceType(IFace));
846 }
847 
848 Sema::ObjCMessageKind Sema::getObjCMessageKind(Scope *S,
849                                                IdentifierInfo *Name,
850                                                SourceLocation NameLoc,
851                                                bool IsSuper,
852                                                bool HasTrailingDot,
853                                                ParsedType &ReceiverType) {
854   ReceiverType = ParsedType();
855 
856   // If the identifier is "super" and there is no trailing dot, we're
857   // messaging super. If the identifier is "super" and there is a
858   // trailing dot, it's an instance message.
859   if (IsSuper && S->isInObjcMethodScope())
860     return HasTrailingDot? ObjCInstanceMessage : ObjCSuperMessage;
861 
862   LookupResult Result(*this, Name, NameLoc, LookupOrdinaryName);
863   LookupName(Result, S);
864 
865   switch (Result.getResultKind()) {
866   case LookupResult::NotFound:
867     // Normal name lookup didn't find anything. If we're in an
868     // Objective-C method, look for ivars. If we find one, we're done!
869     // FIXME: This is a hack. Ivar lookup should be part of normal
870     // lookup.
871     if (ObjCMethodDecl *Method = getCurMethodDecl()) {
872       if (!Method->getClassInterface()) {
873         // Fall back: let the parser try to parse it as an instance message.
874         return ObjCInstanceMessage;
875       }
876 
877       ObjCInterfaceDecl *ClassDeclared;
878       if (Method->getClassInterface()->lookupInstanceVariable(Name,
879                                                               ClassDeclared))
880         return ObjCInstanceMessage;
881     }
882 
883     // Break out; we'll perform typo correction below.
884     break;
885 
886   case LookupResult::NotFoundInCurrentInstantiation:
887   case LookupResult::FoundOverloaded:
888   case LookupResult::FoundUnresolvedValue:
889   case LookupResult::Ambiguous:
890     Result.suppressDiagnostics();
891     return ObjCInstanceMessage;
892 
893   case LookupResult::Found: {
894     // If the identifier is a class or not, and there is a trailing dot,
895     // it's an instance message.
896     if (HasTrailingDot)
897       return ObjCInstanceMessage;
898     // We found something. If it's a type, then we have a class
899     // message. Otherwise, it's an instance message.
900     NamedDecl *ND = Result.getFoundDecl();
901     QualType T;
902     if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(ND))
903       T = Context.getObjCInterfaceType(Class);
904     else if (TypeDecl *Type = dyn_cast<TypeDecl>(ND))
905       T = Context.getTypeDeclType(Type);
906     else
907       return ObjCInstanceMessage;
908 
909     //  We have a class message, and T is the type we're
910     //  messaging. Build source-location information for it.
911     TypeSourceInfo *TSInfo = Context.getTrivialTypeSourceInfo(T, NameLoc);
912     ReceiverType = CreateParsedType(T, TSInfo);
913     return ObjCClassMessage;
914   }
915   }
916 
917   // Determine our typo-correction context.
918   CorrectTypoContext CTC = CTC_Expression;
919   if (ObjCMethodDecl *Method = getCurMethodDecl())
920     if (Method->getClassInterface() &&
921         Method->getClassInterface()->getSuperClass())
922       CTC = CTC_ObjCMessageReceiver;
923 
924   if (TypoCorrection Corrected = CorrectTypo(Result.getLookupNameInfo(),
925                                              Result.getLookupKind(), S, NULL,
926                                              NULL, false, CTC)) {
927     if (NamedDecl *ND = Corrected.getCorrectionDecl()) {
928       // If we found a declaration, correct when it refers to an Objective-C
929       // class.
930       if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(ND)) {
931         Diag(NameLoc, diag::err_unknown_receiver_suggest)
932           << Name << Corrected.getCorrection()
933           << FixItHint::CreateReplacement(SourceRange(NameLoc),
934                                           ND->getNameAsString());
935         Diag(ND->getLocation(), diag::note_previous_decl)
936           << Corrected.getCorrection();
937 
938         QualType T = Context.getObjCInterfaceType(Class);
939         TypeSourceInfo *TSInfo = Context.getTrivialTypeSourceInfo(T, NameLoc);
940         ReceiverType = CreateParsedType(T, TSInfo);
941         return ObjCClassMessage;
942       }
943     } else if (Corrected.isKeyword() &&
944                Corrected.getCorrectionAsIdentifierInfo()->isStr("super")) {
945       // If we've found the keyword "super", this is a send to super.
946       Diag(NameLoc, diag::err_unknown_receiver_suggest)
947         << Name << Corrected.getCorrection()
948         << FixItHint::CreateReplacement(SourceRange(NameLoc), "super");
949       return ObjCSuperMessage;
950     }
951   }
952 
953   // Fall back: let the parser try to parse it as an instance message.
954   return ObjCInstanceMessage;
955 }
956 
957 ExprResult Sema::ActOnSuperMessage(Scope *S,
958                                    SourceLocation SuperLoc,
959                                    Selector Sel,
960                                    SourceLocation LBracLoc,
961                                    ArrayRef<SourceLocation> SelectorLocs,
962                                    SourceLocation RBracLoc,
963                                    MultiExprArg Args) {
964   // Determine whether we are inside a method or not.
965   ObjCMethodDecl *Method = tryCaptureObjCSelf();
966   if (!Method) {
967     Diag(SuperLoc, diag::err_invalid_receiver_to_message_super);
968     return ExprError();
969   }
970 
971   ObjCInterfaceDecl *Class = Method->getClassInterface();
972   if (!Class) {
973     Diag(SuperLoc, diag::error_no_super_class_message)
974       << Method->getDeclName();
975     return ExprError();
976   }
977 
978   ObjCInterfaceDecl *Super = Class->getSuperClass();
979   if (!Super) {
980     // The current class does not have a superclass.
981     Diag(SuperLoc, diag::error_root_class_cannot_use_super)
982       << Class->getIdentifier();
983     return ExprError();
984   }
985 
986   // We are in a method whose class has a superclass, so 'super'
987   // is acting as a keyword.
988   if (Method->isInstanceMethod()) {
989     if (Sel.getMethodFamily() == OMF_dealloc)
990       ObjCShouldCallSuperDealloc = false;
991     if (Sel.getMethodFamily() == OMF_finalize)
992       ObjCShouldCallSuperFinalize = false;
993 
994     // Since we are in an instance method, this is an instance
995     // message to the superclass instance.
996     QualType SuperTy = Context.getObjCInterfaceType(Super);
997     SuperTy = Context.getObjCObjectPointerType(SuperTy);
998     return BuildInstanceMessage(0, SuperTy, SuperLoc,
999                                 Sel, /*Method=*/0,
1000                                 LBracLoc, SelectorLocs, RBracLoc, move(Args));
1001   }
1002 
1003   // Since we are in a class method, this is a class message to
1004   // the superclass.
1005   return BuildClassMessage(/*ReceiverTypeInfo=*/0,
1006                            Context.getObjCInterfaceType(Super),
1007                            SuperLoc, Sel, /*Method=*/0,
1008                            LBracLoc, SelectorLocs, RBracLoc, move(Args));
1009 }
1010 
1011 /// \brief Build an Objective-C class message expression.
1012 ///
1013 /// This routine takes care of both normal class messages and
1014 /// class messages to the superclass.
1015 ///
1016 /// \param ReceiverTypeInfo Type source information that describes the
1017 /// receiver of this message. This may be NULL, in which case we are
1018 /// sending to the superclass and \p SuperLoc must be a valid source
1019 /// location.
1020 
1021 /// \param ReceiverType The type of the object receiving the
1022 /// message. When \p ReceiverTypeInfo is non-NULL, this is the same
1023 /// type as that refers to. For a superclass send, this is the type of
1024 /// the superclass.
1025 ///
1026 /// \param SuperLoc The location of the "super" keyword in a
1027 /// superclass message.
1028 ///
1029 /// \param Sel The selector to which the message is being sent.
1030 ///
1031 /// \param Method The method that this class message is invoking, if
1032 /// already known.
1033 ///
1034 /// \param LBracLoc The location of the opening square bracket ']'.
1035 ///
1036 /// \param RBrac The location of the closing square bracket ']'.
1037 ///
1038 /// \param Args The message arguments.
1039 ExprResult Sema::BuildClassMessage(TypeSourceInfo *ReceiverTypeInfo,
1040                                    QualType ReceiverType,
1041                                    SourceLocation SuperLoc,
1042                                    Selector Sel,
1043                                    ObjCMethodDecl *Method,
1044                                    SourceLocation LBracLoc,
1045                                    ArrayRef<SourceLocation> SelectorLocs,
1046                                    SourceLocation RBracLoc,
1047                                    MultiExprArg ArgsIn) {
1048   SourceLocation Loc = SuperLoc.isValid()? SuperLoc
1049     : ReceiverTypeInfo->getTypeLoc().getSourceRange().getBegin();
1050   if (LBracLoc.isInvalid()) {
1051     Diag(Loc, diag::err_missing_open_square_message_send)
1052       << FixItHint::CreateInsertion(Loc, "[");
1053     LBracLoc = Loc;
1054   }
1055 
1056   if (ReceiverType->isDependentType()) {
1057     // If the receiver type is dependent, we can't type-check anything
1058     // at this point. Build a dependent expression.
1059     unsigned NumArgs = ArgsIn.size();
1060     Expr **Args = reinterpret_cast<Expr **>(ArgsIn.release());
1061     assert(SuperLoc.isInvalid() && "Message to super with dependent type");
1062     return Owned(ObjCMessageExpr::Create(Context, ReceiverType,
1063                                          VK_RValue, LBracLoc, ReceiverTypeInfo,
1064                                          Sel, SelectorLocs, /*Method=*/0,
1065                                          makeArrayRef(Args, NumArgs),RBracLoc));
1066   }
1067 
1068   // Find the class to which we are sending this message.
1069   ObjCInterfaceDecl *Class = 0;
1070   const ObjCObjectType *ClassType = ReceiverType->getAs<ObjCObjectType>();
1071   if (!ClassType || !(Class = ClassType->getInterface())) {
1072     Diag(Loc, diag::err_invalid_receiver_class_message)
1073       << ReceiverType;
1074     return ExprError();
1075   }
1076   assert(Class && "We don't know which class we're messaging?");
1077   // objc++ diagnoses during typename annotation.
1078   if (!getLangOptions().CPlusPlus)
1079     (void)DiagnoseUseOfDecl(Class, Loc);
1080   // Find the method we are messaging.
1081   if (!Method) {
1082     SourceRange TypeRange
1083       = SuperLoc.isValid()? SourceRange(SuperLoc)
1084                           : ReceiverTypeInfo->getTypeLoc().getSourceRange();
1085     if (RequireCompleteType(Loc, Context.getObjCInterfaceType(Class),
1086                             (getLangOptions().ObjCAutoRefCount
1087                                ? PDiag(diag::err_arc_receiver_forward_class)
1088                                : PDiag(diag::warn_receiver_forward_class))
1089                                    << TypeRange)) {
1090       // A forward class used in messaging is treated as a 'Class'
1091       Method = LookupFactoryMethodInGlobalPool(Sel,
1092                                                SourceRange(LBracLoc, RBracLoc));
1093       if (Method && !getLangOptions().ObjCAutoRefCount)
1094         Diag(Method->getLocation(), diag::note_method_sent_forward_class)
1095           << Method->getDeclName();
1096     }
1097     if (!Method)
1098       Method = Class->lookupClassMethod(Sel);
1099 
1100     // If we have an implementation in scope, check "private" methods.
1101     if (!Method)
1102       Method = LookupPrivateClassMethod(Sel, Class);
1103 
1104     if (Method && DiagnoseUseOfDecl(Method, Loc))
1105       return ExprError();
1106   }
1107 
1108   // Check the argument types and determine the result type.
1109   QualType ReturnType;
1110   ExprValueKind VK = VK_RValue;
1111 
1112   unsigned NumArgs = ArgsIn.size();
1113   Expr **Args = reinterpret_cast<Expr **>(ArgsIn.release());
1114   if (CheckMessageArgumentTypes(ReceiverType, Args, NumArgs, Sel, Method, true,
1115                                 SuperLoc.isValid(), LBracLoc, RBracLoc,
1116                                 ReturnType, VK))
1117     return ExprError();
1118 
1119   if (Method && !Method->getResultType()->isVoidType() &&
1120       RequireCompleteType(LBracLoc, Method->getResultType(),
1121                           diag::err_illegal_message_expr_incomplete_type))
1122     return ExprError();
1123 
1124   // Construct the appropriate ObjCMessageExpr.
1125   Expr *Result;
1126   if (SuperLoc.isValid())
1127     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
1128                                      SuperLoc, /*IsInstanceSuper=*/false,
1129                                      ReceiverType, Sel, SelectorLocs,
1130                                      Method, makeArrayRef(Args, NumArgs),
1131                                      RBracLoc);
1132   else
1133     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
1134                                      ReceiverTypeInfo, Sel, SelectorLocs,
1135                                      Method, makeArrayRef(Args, NumArgs),
1136                                      RBracLoc);
1137   return MaybeBindToTemporary(Result);
1138 }
1139 
1140 // ActOnClassMessage - used for both unary and keyword messages.
1141 // ArgExprs is optional - if it is present, the number of expressions
1142 // is obtained from Sel.getNumArgs().
1143 ExprResult Sema::ActOnClassMessage(Scope *S,
1144                                    ParsedType Receiver,
1145                                    Selector Sel,
1146                                    SourceLocation LBracLoc,
1147                                    ArrayRef<SourceLocation> SelectorLocs,
1148                                    SourceLocation RBracLoc,
1149                                    MultiExprArg Args) {
1150   TypeSourceInfo *ReceiverTypeInfo;
1151   QualType ReceiverType = GetTypeFromParser(Receiver, &ReceiverTypeInfo);
1152   if (ReceiverType.isNull())
1153     return ExprError();
1154 
1155 
1156   if (!ReceiverTypeInfo)
1157     ReceiverTypeInfo = Context.getTrivialTypeSourceInfo(ReceiverType, LBracLoc);
1158 
1159   return BuildClassMessage(ReceiverTypeInfo, ReceiverType,
1160                            /*SuperLoc=*/SourceLocation(), Sel, /*Method=*/0,
1161                            LBracLoc, SelectorLocs, RBracLoc, move(Args));
1162 }
1163 
1164 /// \brief Build an Objective-C instance message expression.
1165 ///
1166 /// This routine takes care of both normal instance messages and
1167 /// instance messages to the superclass instance.
1168 ///
1169 /// \param Receiver The expression that computes the object that will
1170 /// receive this message. This may be empty, in which case we are
1171 /// sending to the superclass instance and \p SuperLoc must be a valid
1172 /// source location.
1173 ///
1174 /// \param ReceiverType The (static) type of the object receiving the
1175 /// message. When a \p Receiver expression is provided, this is the
1176 /// same type as that expression. For a superclass instance send, this
1177 /// is a pointer to the type of the superclass.
1178 ///
1179 /// \param SuperLoc The location of the "super" keyword in a
1180 /// superclass instance message.
1181 ///
1182 /// \param Sel The selector to which the message is being sent.
1183 ///
1184 /// \param Method The method that this instance message is invoking, if
1185 /// already known.
1186 ///
1187 /// \param LBracLoc The location of the opening square bracket ']'.
1188 ///
1189 /// \param RBrac The location of the closing square bracket ']'.
1190 ///
1191 /// \param Args The message arguments.
1192 ExprResult Sema::BuildInstanceMessage(Expr *Receiver,
1193                                       QualType ReceiverType,
1194                                       SourceLocation SuperLoc,
1195                                       Selector Sel,
1196                                       ObjCMethodDecl *Method,
1197                                       SourceLocation LBracLoc,
1198                                       ArrayRef<SourceLocation> SelectorLocs,
1199                                       SourceLocation RBracLoc,
1200                                       MultiExprArg ArgsIn) {
1201   // The location of the receiver.
1202   SourceLocation Loc = SuperLoc.isValid()? SuperLoc : Receiver->getLocStart();
1203 
1204   if (LBracLoc.isInvalid()) {
1205     Diag(Loc, diag::err_missing_open_square_message_send)
1206       << FixItHint::CreateInsertion(Loc, "[");
1207     LBracLoc = Loc;
1208   }
1209 
1210   // If we have a receiver expression, perform appropriate promotions
1211   // and determine receiver type.
1212   if (Receiver) {
1213     if (Receiver->hasPlaceholderType()) {
1214       ExprResult Result;
1215       if (Receiver->getType() == Context.UnknownAnyTy)
1216         Result = forceUnknownAnyToType(Receiver, Context.getObjCIdType());
1217       else
1218         Result = CheckPlaceholderExpr(Receiver);
1219       if (Result.isInvalid()) return ExprError();
1220       Receiver = Result.take();
1221     }
1222 
1223     if (Receiver->isTypeDependent()) {
1224       // If the receiver is type-dependent, we can't type-check anything
1225       // at this point. Build a dependent expression.
1226       unsigned NumArgs = ArgsIn.size();
1227       Expr **Args = reinterpret_cast<Expr **>(ArgsIn.release());
1228       assert(SuperLoc.isInvalid() && "Message to super with dependent type");
1229       return Owned(ObjCMessageExpr::Create(Context, Context.DependentTy,
1230                                            VK_RValue, LBracLoc, Receiver, Sel,
1231                                            SelectorLocs, /*Method=*/0,
1232                                            makeArrayRef(Args, NumArgs),
1233                                            RBracLoc));
1234     }
1235 
1236     // If necessary, apply function/array conversion to the receiver.
1237     // C99 6.7.5.3p[7,8].
1238     ExprResult Result = DefaultFunctionArrayLvalueConversion(Receiver);
1239     if (Result.isInvalid())
1240       return ExprError();
1241     Receiver = Result.take();
1242     ReceiverType = Receiver->getType();
1243   }
1244 
1245   if (!Method) {
1246     // Handle messages to id.
1247     bool receiverIsId = ReceiverType->isObjCIdType();
1248     if (receiverIsId || ReceiverType->isBlockPointerType() ||
1249         (Receiver && Context.isObjCNSObjectType(Receiver->getType()))) {
1250       Method = LookupInstanceMethodInGlobalPool(Sel,
1251                                                 SourceRange(LBracLoc, RBracLoc),
1252                                                 receiverIsId);
1253       if (!Method)
1254         Method = LookupFactoryMethodInGlobalPool(Sel,
1255                                                  SourceRange(LBracLoc, RBracLoc),
1256                                                  receiverIsId);
1257       if (Method)
1258         DiagnoseAvailabilityOfDecl(Method, Loc, 0);
1259 
1260     } else if (ReceiverType->isObjCClassType() ||
1261                ReceiverType->isObjCQualifiedClassType()) {
1262       // Handle messages to Class.
1263       // We allow sending a message to a qualified Class ("Class<foo>"), which
1264       // is ok as long as one of the protocols implements the selector (if not, warn).
1265       if (const ObjCObjectPointerType *QClassTy
1266             = ReceiverType->getAsObjCQualifiedClassType()) {
1267         // Search protocols for class methods.
1268         Method = LookupMethodInQualifiedType(Sel, QClassTy, false);
1269         if (!Method) {
1270           Method = LookupMethodInQualifiedType(Sel, QClassTy, true);
1271           // warn if instance method found for a Class message.
1272           if (Method) {
1273             Diag(Loc, diag::warn_instance_method_on_class_found)
1274               << Method->getSelector() << Sel;
1275             Diag(Method->getLocation(), diag::note_method_declared_at);
1276           }
1277         }
1278       } else {
1279         if (ObjCMethodDecl *CurMeth = getCurMethodDecl()) {
1280           if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface()) {
1281             // First check the public methods in the class interface.
1282             Method = ClassDecl->lookupClassMethod(Sel);
1283 
1284             if (!Method)
1285               Method = LookupPrivateClassMethod(Sel, ClassDecl);
1286           }
1287           if (Method && DiagnoseUseOfDecl(Method, Loc))
1288             return ExprError();
1289         }
1290         if (!Method) {
1291           // If not messaging 'self', look for any factory method named 'Sel'.
1292           if (!Receiver || !isSelfExpr(Receiver)) {
1293             Method = LookupFactoryMethodInGlobalPool(Sel,
1294                                                 SourceRange(LBracLoc, RBracLoc),
1295                                                      true);
1296             if (!Method) {
1297               // If no class (factory) method was found, check if an _instance_
1298               // method of the same name exists in the root class only.
1299               Method = LookupInstanceMethodInGlobalPool(Sel,
1300                                                SourceRange(LBracLoc, RBracLoc),
1301                                                         true);
1302               if (Method)
1303                   if (const ObjCInterfaceDecl *ID =
1304                       dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext())) {
1305                     if (ID->getSuperClass())
1306                       Diag(Loc, diag::warn_root_inst_method_not_found)
1307                       << Sel << SourceRange(LBracLoc, RBracLoc);
1308                   }
1309             }
1310           }
1311         }
1312       }
1313     } else {
1314       ObjCInterfaceDecl* ClassDecl = 0;
1315 
1316       // We allow sending a message to a qualified ID ("id<foo>"), which is ok as
1317       // long as one of the protocols implements the selector (if not, warn).
1318       if (const ObjCObjectPointerType *QIdTy
1319                                    = ReceiverType->getAsObjCQualifiedIdType()) {
1320         // Search protocols for instance methods.
1321         Method = LookupMethodInQualifiedType(Sel, QIdTy, true);
1322         if (!Method)
1323           Method = LookupMethodInQualifiedType(Sel, QIdTy, false);
1324       } else if (const ObjCObjectPointerType *OCIType
1325                    = ReceiverType->getAsObjCInterfacePointerType()) {
1326         // We allow sending a message to a pointer to an interface (an object).
1327         ClassDecl = OCIType->getInterfaceDecl();
1328 
1329         // Try to complete the type. Under ARC, this is a hard error from which
1330         // we don't try to recover.
1331         const ObjCInterfaceDecl *forwardClass = 0;
1332         if (RequireCompleteType(Loc, OCIType->getPointeeType(),
1333               getLangOptions().ObjCAutoRefCount
1334                 ? PDiag(diag::err_arc_receiver_forward_instance)
1335                     << (Receiver ? Receiver->getSourceRange()
1336                                  : SourceRange(SuperLoc))
1337                 : PDiag())) {
1338           if (getLangOptions().ObjCAutoRefCount)
1339             return ExprError();
1340 
1341           forwardClass = OCIType->getInterfaceDecl();
1342         }
1343 
1344         // FIXME: consider using LookupInstanceMethodInGlobalPool, since it will be
1345         // faster than the following method (which can do *many* linear searches).
1346         // The idea is to add class info to MethodPool.
1347         Method = ClassDecl->lookupInstanceMethod(Sel);
1348 
1349         if (!Method)
1350           // Search protocol qualifiers.
1351           Method = LookupMethodInQualifiedType(Sel, OCIType, true);
1352 
1353         if (!Method) {
1354           // If we have implementations in scope, check "private" methods.
1355           Method = LookupPrivateInstanceMethod(Sel, ClassDecl);
1356 
1357           if (!Method && getLangOptions().ObjCAutoRefCount) {
1358             Diag(Loc, diag::err_arc_may_not_respond)
1359               << OCIType->getPointeeType() << Sel;
1360             return ExprError();
1361           }
1362 
1363           if (!Method && (!Receiver || !isSelfExpr(Receiver))) {
1364             // If we still haven't found a method, look in the global pool. This
1365             // behavior isn't very desirable, however we need it for GCC
1366             // compatibility. FIXME: should we deviate??
1367             if (OCIType->qual_empty()) {
1368               Method = LookupInstanceMethodInGlobalPool(Sel,
1369                                                  SourceRange(LBracLoc, RBracLoc));
1370               if (Method && !forwardClass)
1371                 Diag(Loc, diag::warn_maynot_respond)
1372                   << OCIType->getInterfaceDecl()->getIdentifier() << Sel;
1373             }
1374           }
1375         }
1376         if (Method && DiagnoseUseOfDecl(Method, Loc, forwardClass))
1377           return ExprError();
1378       } else if (!getLangOptions().ObjCAutoRefCount &&
1379                  !Context.getObjCIdType().isNull() &&
1380                  (ReceiverType->isPointerType() ||
1381                   ReceiverType->isIntegerType())) {
1382         // Implicitly convert integers and pointers to 'id' but emit a warning.
1383         // But not in ARC.
1384         Diag(Loc, diag::warn_bad_receiver_type)
1385           << ReceiverType
1386           << Receiver->getSourceRange();
1387         if (ReceiverType->isPointerType())
1388           Receiver = ImpCastExprToType(Receiver, Context.getObjCIdType(),
1389                             CK_CPointerToObjCPointerCast).take();
1390         else {
1391           // TODO: specialized warning on null receivers?
1392           bool IsNull = Receiver->isNullPointerConstant(Context,
1393                                               Expr::NPC_ValueDependentIsNull);
1394           Receiver = ImpCastExprToType(Receiver, Context.getObjCIdType(),
1395                             IsNull ? CK_NullToPointer : CK_IntegralToPointer).take();
1396         }
1397         ReceiverType = Receiver->getType();
1398       } else {
1399         ExprResult ReceiverRes;
1400         if (getLangOptions().CPlusPlus)
1401           ReceiverRes = PerformContextuallyConvertToObjCPointer(Receiver);
1402         if (ReceiverRes.isUsable()) {
1403           Receiver = ReceiverRes.take();
1404           return BuildInstanceMessage(Receiver,
1405                                       ReceiverType,
1406                                       SuperLoc,
1407                                       Sel,
1408                                       Method,
1409                                       LBracLoc,
1410                                       SelectorLocs,
1411                                       RBracLoc,
1412                                       move(ArgsIn));
1413         } else {
1414           // Reject other random receiver types (e.g. structs).
1415           Diag(Loc, diag::err_bad_receiver_type)
1416             << ReceiverType << Receiver->getSourceRange();
1417           return ExprError();
1418         }
1419       }
1420     }
1421   }
1422 
1423   // Check the message arguments.
1424   unsigned NumArgs = ArgsIn.size();
1425   Expr **Args = reinterpret_cast<Expr **>(ArgsIn.release());
1426   QualType ReturnType;
1427   ExprValueKind VK = VK_RValue;
1428   bool ClassMessage = (ReceiverType->isObjCClassType() ||
1429                        ReceiverType->isObjCQualifiedClassType());
1430   if (CheckMessageArgumentTypes(ReceiverType, Args, NumArgs, Sel, Method,
1431                                 ClassMessage, SuperLoc.isValid(),
1432                                 LBracLoc, RBracLoc, ReturnType, VK))
1433     return ExprError();
1434 
1435   if (Method && !Method->getResultType()->isVoidType() &&
1436       RequireCompleteType(LBracLoc, Method->getResultType(),
1437                           diag::err_illegal_message_expr_incomplete_type))
1438     return ExprError();
1439 
1440   SourceLocation SelLoc = SelectorLocs.front();
1441 
1442   // In ARC, forbid the user from sending messages to
1443   // retain/release/autorelease/dealloc/retainCount explicitly.
1444   if (getLangOptions().ObjCAutoRefCount) {
1445     ObjCMethodFamily family =
1446       (Method ? Method->getMethodFamily() : Sel.getMethodFamily());
1447     switch (family) {
1448     case OMF_init:
1449       if (Method)
1450         checkInitMethod(Method, ReceiverType);
1451 
1452     case OMF_None:
1453     case OMF_alloc:
1454     case OMF_copy:
1455     case OMF_finalize:
1456     case OMF_mutableCopy:
1457     case OMF_new:
1458     case OMF_self:
1459       break;
1460 
1461     case OMF_dealloc:
1462     case OMF_retain:
1463     case OMF_release:
1464     case OMF_autorelease:
1465     case OMF_retainCount:
1466       Diag(Loc, diag::err_arc_illegal_explicit_message)
1467         << Sel << SelLoc;
1468       break;
1469 
1470     case OMF_performSelector:
1471       if (Method && NumArgs >= 1) {
1472         if (ObjCSelectorExpr *SelExp = dyn_cast<ObjCSelectorExpr>(Args[0])) {
1473           Selector ArgSel = SelExp->getSelector();
1474           ObjCMethodDecl *SelMethod =
1475             LookupInstanceMethodInGlobalPool(ArgSel,
1476                                              SelExp->getSourceRange());
1477           if (!SelMethod)
1478             SelMethod =
1479               LookupFactoryMethodInGlobalPool(ArgSel,
1480                                               SelExp->getSourceRange());
1481           if (SelMethod) {
1482             ObjCMethodFamily SelFamily = SelMethod->getMethodFamily();
1483             switch (SelFamily) {
1484               case OMF_alloc:
1485               case OMF_copy:
1486               case OMF_mutableCopy:
1487               case OMF_new:
1488               case OMF_self:
1489               case OMF_init:
1490                 // Issue error, unless ns_returns_not_retained.
1491                 if (!SelMethod->hasAttr<NSReturnsNotRetainedAttr>()) {
1492                   // selector names a +1 method
1493                   Diag(SelLoc,
1494                        diag::err_arc_perform_selector_retains);
1495                   Diag(SelMethod->getLocation(), diag::note_method_declared_at);
1496                 }
1497                 break;
1498               default:
1499                 // +0 call. OK. unless ns_returns_retained.
1500                 if (SelMethod->hasAttr<NSReturnsRetainedAttr>()) {
1501                   // selector names a +1 method
1502                   Diag(SelLoc,
1503                        diag::err_arc_perform_selector_retains);
1504                   Diag(SelMethod->getLocation(), diag::note_method_declared_at);
1505                 }
1506                 break;
1507             }
1508           }
1509         } else {
1510           // error (may leak).
1511           Diag(SelLoc, diag::warn_arc_perform_selector_leaks);
1512           Diag(Args[0]->getExprLoc(), diag::note_used_here);
1513         }
1514       }
1515       break;
1516     }
1517   }
1518 
1519   // Construct the appropriate ObjCMessageExpr instance.
1520   ObjCMessageExpr *Result;
1521   if (SuperLoc.isValid())
1522     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
1523                                      SuperLoc,  /*IsInstanceSuper=*/true,
1524                                      ReceiverType, Sel, SelectorLocs, Method,
1525                                      makeArrayRef(Args, NumArgs), RBracLoc);
1526   else
1527     Result = ObjCMessageExpr::Create(Context, ReturnType, VK, LBracLoc,
1528                                      Receiver, Sel, SelectorLocs, Method,
1529                                      makeArrayRef(Args, NumArgs), RBracLoc);
1530 
1531   if (getLangOptions().ObjCAutoRefCount) {
1532     // In ARC, annotate delegate init calls.
1533     if (Result->getMethodFamily() == OMF_init &&
1534         (SuperLoc.isValid() || isSelfExpr(Receiver))) {
1535       // Only consider init calls *directly* in init implementations,
1536       // not within blocks.
1537       ObjCMethodDecl *method = dyn_cast<ObjCMethodDecl>(CurContext);
1538       if (method && method->getMethodFamily() == OMF_init) {
1539         // The implicit assignment to self means we also don't want to
1540         // consume the result.
1541         Result->setDelegateInitCall(true);
1542         return Owned(Result);
1543       }
1544     }
1545 
1546     // In ARC, check for message sends which are likely to introduce
1547     // retain cycles.
1548     checkRetainCycles(Result);
1549   }
1550 
1551   return MaybeBindToTemporary(Result);
1552 }
1553 
1554 // ActOnInstanceMessage - used for both unary and keyword messages.
1555 // ArgExprs is optional - if it is present, the number of expressions
1556 // is obtained from Sel.getNumArgs().
1557 ExprResult Sema::ActOnInstanceMessage(Scope *S,
1558                                       Expr *Receiver,
1559                                       Selector Sel,
1560                                       SourceLocation LBracLoc,
1561                                       ArrayRef<SourceLocation> SelectorLocs,
1562                                       SourceLocation RBracLoc,
1563                                       MultiExprArg Args) {
1564   if (!Receiver)
1565     return ExprError();
1566 
1567   return BuildInstanceMessage(Receiver, Receiver->getType(),
1568                               /*SuperLoc=*/SourceLocation(), Sel, /*Method=*/0,
1569                               LBracLoc, SelectorLocs, RBracLoc, move(Args));
1570 }
1571 
1572 enum ARCConversionTypeClass {
1573   /// int, void, struct A
1574   ACTC_none,
1575 
1576   /// id, void (^)()
1577   ACTC_retainable,
1578 
1579   /// id*, id***, void (^*)(),
1580   ACTC_indirectRetainable,
1581 
1582   /// void* might be a normal C type, or it might a CF type.
1583   ACTC_voidPtr,
1584 
1585   /// struct A*
1586   ACTC_coreFoundation
1587 };
1588 static bool isAnyRetainable(ARCConversionTypeClass ACTC) {
1589   return (ACTC == ACTC_retainable ||
1590           ACTC == ACTC_coreFoundation ||
1591           ACTC == ACTC_voidPtr);
1592 }
1593 static bool isAnyCLike(ARCConversionTypeClass ACTC) {
1594   return ACTC == ACTC_none ||
1595          ACTC == ACTC_voidPtr ||
1596          ACTC == ACTC_coreFoundation;
1597 }
1598 
1599 static ARCConversionTypeClass classifyTypeForARCConversion(QualType type) {
1600   bool isIndirect = false;
1601 
1602   // Ignore an outermost reference type.
1603   if (const ReferenceType *ref = type->getAs<ReferenceType>()) {
1604     type = ref->getPointeeType();
1605     isIndirect = true;
1606   }
1607 
1608   // Drill through pointers and arrays recursively.
1609   while (true) {
1610     if (const PointerType *ptr = type->getAs<PointerType>()) {
1611       type = ptr->getPointeeType();
1612 
1613       // The first level of pointer may be the innermost pointer on a CF type.
1614       if (!isIndirect) {
1615         if (type->isVoidType()) return ACTC_voidPtr;
1616         if (type->isRecordType()) return ACTC_coreFoundation;
1617       }
1618     } else if (const ArrayType *array = type->getAsArrayTypeUnsafe()) {
1619       type = QualType(array->getElementType()->getBaseElementTypeUnsafe(), 0);
1620     } else {
1621       break;
1622     }
1623     isIndirect = true;
1624   }
1625 
1626   if (isIndirect) {
1627     if (type->isObjCARCBridgableType())
1628       return ACTC_indirectRetainable;
1629     return ACTC_none;
1630   }
1631 
1632   if (type->isObjCARCBridgableType())
1633     return ACTC_retainable;
1634 
1635   return ACTC_none;
1636 }
1637 
1638 namespace {
1639   /// A result from the cast checker.
1640   enum ACCResult {
1641     /// Cannot be casted.
1642     ACC_invalid,
1643 
1644     /// Can be safely retained or not retained.
1645     ACC_bottom,
1646 
1647     /// Can be casted at +0.
1648     ACC_plusZero,
1649 
1650     /// Can be casted at +1.
1651     ACC_plusOne
1652   };
1653   ACCResult merge(ACCResult left, ACCResult right) {
1654     if (left == right) return left;
1655     if (left == ACC_bottom) return right;
1656     if (right == ACC_bottom) return left;
1657     return ACC_invalid;
1658   }
1659 
1660   /// A checker which white-lists certain expressions whose conversion
1661   /// to or from retainable type would otherwise be forbidden in ARC.
1662   class ARCCastChecker : public StmtVisitor<ARCCastChecker, ACCResult> {
1663     typedef StmtVisitor<ARCCastChecker, ACCResult> super;
1664 
1665     ASTContext &Context;
1666     ARCConversionTypeClass SourceClass;
1667     ARCConversionTypeClass TargetClass;
1668 
1669     static bool isCFType(QualType type) {
1670       // Someday this can use ns_bridged.  For now, it has to do this.
1671       return type->isCARCBridgableType();
1672     }
1673 
1674   public:
1675     ARCCastChecker(ASTContext &Context, ARCConversionTypeClass source,
1676                    ARCConversionTypeClass target)
1677       : Context(Context), SourceClass(source), TargetClass(target) {}
1678 
1679     using super::Visit;
1680     ACCResult Visit(Expr *e) {
1681       return super::Visit(e->IgnoreParens());
1682     }
1683 
1684     ACCResult VisitStmt(Stmt *s) {
1685       return ACC_invalid;
1686     }
1687 
1688     /// Null pointer constants can be casted however you please.
1689     ACCResult VisitExpr(Expr *e) {
1690       if (e->isNullPointerConstant(Context, Expr::NPC_ValueDependentIsNotNull))
1691         return ACC_bottom;
1692       return ACC_invalid;
1693     }
1694 
1695     /// Objective-C string literals can be safely casted.
1696     ACCResult VisitObjCStringLiteral(ObjCStringLiteral *e) {
1697       // If we're casting to any retainable type, go ahead.  Global
1698       // strings are immune to retains, so this is bottom.
1699       if (isAnyRetainable(TargetClass)) return ACC_bottom;
1700 
1701       return ACC_invalid;
1702     }
1703 
1704     /// Look through certain implicit and explicit casts.
1705     ACCResult VisitCastExpr(CastExpr *e) {
1706       switch (e->getCastKind()) {
1707         case CK_NullToPointer:
1708           return ACC_bottom;
1709 
1710         case CK_NoOp:
1711         case CK_LValueToRValue:
1712         case CK_BitCast:
1713         case CK_CPointerToObjCPointerCast:
1714         case CK_BlockPointerToObjCPointerCast:
1715         case CK_AnyPointerToBlockPointerCast:
1716           return Visit(e->getSubExpr());
1717 
1718         default:
1719           return ACC_invalid;
1720       }
1721     }
1722 
1723     /// Look through unary extension.
1724     ACCResult VisitUnaryExtension(UnaryOperator *e) {
1725       return Visit(e->getSubExpr());
1726     }
1727 
1728     /// Ignore the LHS of a comma operator.
1729     ACCResult VisitBinComma(BinaryOperator *e) {
1730       return Visit(e->getRHS());
1731     }
1732 
1733     /// Conditional operators are okay if both sides are okay.
1734     ACCResult VisitConditionalOperator(ConditionalOperator *e) {
1735       ACCResult left = Visit(e->getTrueExpr());
1736       if (left == ACC_invalid) return ACC_invalid;
1737       return merge(left, Visit(e->getFalseExpr()));
1738     }
1739 
1740     /// Look through pseudo-objects.
1741     ACCResult VisitPseudoObjectExpr(PseudoObjectExpr *e) {
1742       // If we're getting here, we should always have a result.
1743       return Visit(e->getResultExpr());
1744     }
1745 
1746     /// Statement expressions are okay if their result expression is okay.
1747     ACCResult VisitStmtExpr(StmtExpr *e) {
1748       return Visit(e->getSubStmt()->body_back());
1749     }
1750 
1751     /// Some declaration references are okay.
1752     ACCResult VisitDeclRefExpr(DeclRefExpr *e) {
1753       // References to global constants from system headers are okay.
1754       // These are things like 'kCFStringTransformToLatin'.  They are
1755       // can also be assumed to be immune to retains.
1756       VarDecl *var = dyn_cast<VarDecl>(e->getDecl());
1757       if (isAnyRetainable(TargetClass) &&
1758           isAnyRetainable(SourceClass) &&
1759           var &&
1760           var->getStorageClass() == SC_Extern &&
1761           var->getType().isConstQualified() &&
1762           Context.getSourceManager().isInSystemHeader(var->getLocation())) {
1763         return ACC_bottom;
1764       }
1765 
1766       // Nothing else.
1767       return ACC_invalid;
1768     }
1769 
1770     /// Some calls are okay.
1771     ACCResult VisitCallExpr(CallExpr *e) {
1772       if (FunctionDecl *fn = e->getDirectCallee())
1773         if (ACCResult result = checkCallToFunction(fn))
1774           return result;
1775 
1776       return super::VisitCallExpr(e);
1777     }
1778 
1779     ACCResult checkCallToFunction(FunctionDecl *fn) {
1780       // Require a CF*Ref return type.
1781       if (!isCFType(fn->getResultType()))
1782         return ACC_invalid;
1783 
1784       if (!isAnyRetainable(TargetClass))
1785         return ACC_invalid;
1786 
1787       // Honor an explicit 'not retained' attribute.
1788       if (fn->hasAttr<CFReturnsNotRetainedAttr>())
1789         return ACC_plusZero;
1790 
1791       // Honor an explicit 'retained' attribute, except that for
1792       // now we're not going to permit implicit handling of +1 results,
1793       // because it's a bit frightening.
1794       if (fn->hasAttr<CFReturnsRetainedAttr>())
1795         return ACC_invalid; // ACC_plusOne if we start accepting this
1796 
1797       // Recognize this specific builtin function, which is used by CFSTR.
1798       unsigned builtinID = fn->getBuiltinID();
1799       if (builtinID == Builtin::BI__builtin___CFStringMakeConstantString)
1800         return ACC_bottom;
1801 
1802       // Otherwise, don't do anything implicit with an unaudited function.
1803       if (!fn->hasAttr<CFAuditedTransferAttr>())
1804         return ACC_invalid;
1805 
1806       // Otherwise, it's +0 unless it follows the create convention.
1807       if (ento::coreFoundation::followsCreateRule(fn))
1808         return ACC_invalid; // ACC_plusOne if we start accepting this
1809 
1810       return ACC_plusZero;
1811     }
1812 
1813     ACCResult VisitObjCMessageExpr(ObjCMessageExpr *e) {
1814       return checkCallToMethod(e->getMethodDecl());
1815     }
1816 
1817     ACCResult VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *e) {
1818       ObjCMethodDecl *method;
1819       if (e->isExplicitProperty())
1820         method = e->getExplicitProperty()->getGetterMethodDecl();
1821       else
1822         method = e->getImplicitPropertyGetter();
1823       return checkCallToMethod(method);
1824     }
1825 
1826     ACCResult checkCallToMethod(ObjCMethodDecl *method) {
1827       if (!method) return ACC_invalid;
1828 
1829       // Check for message sends to functions returning CF types.  We
1830       // just obey the Cocoa conventions with these, even though the
1831       // return type is CF.
1832       if (!isAnyRetainable(TargetClass) || !isCFType(method->getResultType()))
1833         return ACC_invalid;
1834 
1835       // If the method is explicitly marked not-retained, it's +0.
1836       if (method->hasAttr<CFReturnsNotRetainedAttr>())
1837         return ACC_plusZero;
1838 
1839       // If the method is explicitly marked as returning retained, or its
1840       // selector follows a +1 Cocoa convention, treat it as +1.
1841       if (method->hasAttr<CFReturnsRetainedAttr>())
1842         return ACC_plusOne;
1843 
1844       switch (method->getSelector().getMethodFamily()) {
1845       case OMF_alloc:
1846       case OMF_copy:
1847       case OMF_mutableCopy:
1848       case OMF_new:
1849         return ACC_plusOne;
1850 
1851       default:
1852         // Otherwise, treat it as +0.
1853         return ACC_plusZero;
1854       }
1855     }
1856   };
1857 }
1858 
1859 static void
1860 diagnoseObjCARCConversion(Sema &S, SourceRange castRange,
1861                           QualType castType, ARCConversionTypeClass castACTC,
1862                           Expr *castExpr, ARCConversionTypeClass exprACTC,
1863                           Sema::CheckedConversionKind CCK) {
1864   SourceLocation loc =
1865     (castRange.isValid() ? castRange.getBegin() : castExpr->getExprLoc());
1866 
1867   if (S.makeUnavailableInSystemHeader(loc,
1868                 "converts between Objective-C and C pointers in -fobjc-arc"))
1869     return;
1870 
1871   QualType castExprType = castExpr->getType();
1872 
1873   unsigned srcKind = 0;
1874   switch (exprACTC) {
1875   case ACTC_none:
1876   case ACTC_coreFoundation:
1877   case ACTC_voidPtr:
1878     srcKind = (castExprType->isPointerType() ? 1 : 0);
1879     break;
1880   case ACTC_retainable:
1881     srcKind = (castExprType->isBlockPointerType() ? 2 : 3);
1882     break;
1883   case ACTC_indirectRetainable:
1884     srcKind = 4;
1885     break;
1886   }
1887 
1888   // Check whether this could be fixed with a bridge cast.
1889   SourceLocation afterLParen = S.PP.getLocForEndOfToken(castRange.getBegin());
1890   SourceLocation noteLoc = afterLParen.isValid() ? afterLParen : loc;
1891 
1892   // Bridge from an ARC type to a CF type.
1893   if (castACTC == ACTC_retainable && isAnyRetainable(exprACTC)) {
1894     S.Diag(loc, diag::err_arc_cast_requires_bridge)
1895       << unsigned(CCK == Sema::CCK_ImplicitConversion) // cast|implicit
1896       << 2 // of C pointer type
1897       << castExprType
1898       << unsigned(castType->isBlockPointerType()) // to ObjC|block type
1899       << castType
1900       << castRange
1901       << castExpr->getSourceRange();
1902 
1903     S.Diag(noteLoc, diag::note_arc_bridge)
1904       << (CCK != Sema::CCK_CStyleCast ? FixItHint() :
1905             FixItHint::CreateInsertion(afterLParen, "__bridge "));
1906     S.Diag(noteLoc, diag::note_arc_bridge_transfer)
1907       << castExprType
1908       << (CCK != Sema::CCK_CStyleCast ? FixItHint() :
1909             FixItHint::CreateInsertion(afterLParen, "__bridge_transfer "));
1910 
1911     return;
1912   }
1913 
1914   // Bridge from a CF type to an ARC type.
1915   if (exprACTC == ACTC_retainable && isAnyRetainable(castACTC)) {
1916     S.Diag(loc, diag::err_arc_cast_requires_bridge)
1917       << unsigned(CCK == Sema::CCK_ImplicitConversion) // cast|implicit
1918       << unsigned(castExprType->isBlockPointerType()) // of ObjC|block type
1919       << castExprType
1920       << 2 // to C pointer type
1921       << castType
1922       << castRange
1923       << castExpr->getSourceRange();
1924 
1925     S.Diag(noteLoc, diag::note_arc_bridge)
1926       << (CCK != Sema::CCK_CStyleCast ? FixItHint() :
1927             FixItHint::CreateInsertion(afterLParen, "__bridge "));
1928     S.Diag(noteLoc, diag::note_arc_bridge_retained)
1929       << castType
1930       << (CCK != Sema::CCK_CStyleCast ? FixItHint() :
1931             FixItHint::CreateInsertion(afterLParen, "__bridge_retained "));
1932 
1933     return;
1934   }
1935 
1936   S.Diag(loc, diag::err_arc_mismatched_cast)
1937     << (CCK != Sema::CCK_ImplicitConversion)
1938     << srcKind << castExprType << castType
1939     << castRange << castExpr->getSourceRange();
1940 }
1941 
1942 Sema::ARCConversionResult
1943 Sema::CheckObjCARCConversion(SourceRange castRange, QualType castType,
1944                              Expr *&castExpr, CheckedConversionKind CCK) {
1945   QualType castExprType = castExpr->getType();
1946 
1947   // For the purposes of the classification, we assume reference types
1948   // will bind to temporaries.
1949   QualType effCastType = castType;
1950   if (const ReferenceType *ref = castType->getAs<ReferenceType>())
1951     effCastType = ref->getPointeeType();
1952 
1953   ARCConversionTypeClass exprACTC = classifyTypeForARCConversion(castExprType);
1954   ARCConversionTypeClass castACTC = classifyTypeForARCConversion(effCastType);
1955   if (exprACTC == castACTC) {
1956     // check for viablity and report error if casting an rvalue to a
1957     // life-time qualifier.
1958     if ((castACTC == ACTC_retainable) &&
1959         (CCK == CCK_CStyleCast || CCK == CCK_OtherCast) &&
1960         (castType != castExprType)) {
1961       const Type *DT = castType.getTypePtr();
1962       QualType QDT = castType;
1963       // We desugar some types but not others. We ignore those
1964       // that cannot happen in a cast; i.e. auto, and those which
1965       // should not be de-sugared; i.e typedef.
1966       if (const ParenType *PT = dyn_cast<ParenType>(DT))
1967         QDT = PT->desugar();
1968       else if (const TypeOfType *TP = dyn_cast<TypeOfType>(DT))
1969         QDT = TP->desugar();
1970       else if (const AttributedType *AT = dyn_cast<AttributedType>(DT))
1971         QDT = AT->desugar();
1972       if (QDT != castType &&
1973           QDT.getObjCLifetime() !=  Qualifiers::OCL_None) {
1974         SourceLocation loc =
1975           (castRange.isValid() ? castRange.getBegin()
1976                               : castExpr->getExprLoc());
1977         Diag(loc, diag::err_arc_nolifetime_behavior);
1978       }
1979     }
1980     return ACR_okay;
1981   }
1982 
1983   if (isAnyCLike(exprACTC) && isAnyCLike(castACTC)) return ACR_okay;
1984 
1985   // Allow all of these types to be cast to integer types (but not
1986   // vice-versa).
1987   if (castACTC == ACTC_none && castType->isIntegralType(Context))
1988     return ACR_okay;
1989 
1990   // Allow casts between pointers to lifetime types (e.g., __strong id*)
1991   // and pointers to void (e.g., cv void *). Casting from void* to lifetime*
1992   // must be explicit.
1993   if (exprACTC == ACTC_indirectRetainable && castACTC == ACTC_voidPtr)
1994     return ACR_okay;
1995   if (castACTC == ACTC_indirectRetainable && exprACTC == ACTC_voidPtr &&
1996       CCK != CCK_ImplicitConversion)
1997     return ACR_okay;
1998 
1999   switch (ARCCastChecker(Context, exprACTC, castACTC).Visit(castExpr)) {
2000   // For invalid casts, fall through.
2001   case ACC_invalid:
2002     break;
2003 
2004   // Do nothing for both bottom and +0.
2005   case ACC_bottom:
2006   case ACC_plusZero:
2007     return ACR_okay;
2008 
2009   // If the result is +1, consume it here.
2010   case ACC_plusOne:
2011     castExpr = ImplicitCastExpr::Create(Context, castExpr->getType(),
2012                                         CK_ARCConsumeObject, castExpr,
2013                                         0, VK_RValue);
2014     ExprNeedsCleanups = true;
2015     return ACR_okay;
2016   }
2017 
2018   // If this is a non-implicit cast from id or block type to a
2019   // CoreFoundation type, delay complaining in case the cast is used
2020   // in an acceptable context.
2021   if (exprACTC == ACTC_retainable && isAnyRetainable(castACTC) &&
2022       CCK != CCK_ImplicitConversion)
2023     return ACR_unbridged;
2024 
2025   diagnoseObjCARCConversion(*this, castRange, castType, castACTC,
2026                             castExpr, exprACTC, CCK);
2027   return ACR_okay;
2028 }
2029 
2030 /// Given that we saw an expression with the ARCUnbridgedCastTy
2031 /// placeholder type, complain bitterly.
2032 void Sema::diagnoseARCUnbridgedCast(Expr *e) {
2033   // We expect the spurious ImplicitCastExpr to already have been stripped.
2034   assert(!e->hasPlaceholderType(BuiltinType::ARCUnbridgedCast));
2035   CastExpr *realCast = cast<CastExpr>(e->IgnoreParens());
2036 
2037   SourceRange castRange;
2038   QualType castType;
2039   CheckedConversionKind CCK;
2040 
2041   if (CStyleCastExpr *cast = dyn_cast<CStyleCastExpr>(realCast)) {
2042     castRange = SourceRange(cast->getLParenLoc(), cast->getRParenLoc());
2043     castType = cast->getTypeAsWritten();
2044     CCK = CCK_CStyleCast;
2045   } else if (ExplicitCastExpr *cast = dyn_cast<ExplicitCastExpr>(realCast)) {
2046     castRange = cast->getTypeInfoAsWritten()->getTypeLoc().getSourceRange();
2047     castType = cast->getTypeAsWritten();
2048     CCK = CCK_OtherCast;
2049   } else {
2050     castType = cast->getType();
2051     CCK = CCK_ImplicitConversion;
2052   }
2053 
2054   ARCConversionTypeClass castACTC =
2055     classifyTypeForARCConversion(castType.getNonReferenceType());
2056 
2057   Expr *castExpr = realCast->getSubExpr();
2058   assert(classifyTypeForARCConversion(castExpr->getType()) == ACTC_retainable);
2059 
2060   diagnoseObjCARCConversion(*this, castRange, castType, castACTC,
2061                             castExpr, ACTC_retainable, CCK);
2062 }
2063 
2064 /// stripARCUnbridgedCast - Given an expression of ARCUnbridgedCast
2065 /// type, remove the placeholder cast.
2066 Expr *Sema::stripARCUnbridgedCast(Expr *e) {
2067   assert(e->hasPlaceholderType(BuiltinType::ARCUnbridgedCast));
2068 
2069   if (ParenExpr *pe = dyn_cast<ParenExpr>(e)) {
2070     Expr *sub = stripARCUnbridgedCast(pe->getSubExpr());
2071     return new (Context) ParenExpr(pe->getLParen(), pe->getRParen(), sub);
2072   } else if (UnaryOperator *uo = dyn_cast<UnaryOperator>(e)) {
2073     assert(uo->getOpcode() == UO_Extension);
2074     Expr *sub = stripARCUnbridgedCast(uo->getSubExpr());
2075     return new (Context) UnaryOperator(sub, UO_Extension, sub->getType(),
2076                                    sub->getValueKind(), sub->getObjectKind(),
2077                                        uo->getOperatorLoc());
2078   } else if (GenericSelectionExpr *gse = dyn_cast<GenericSelectionExpr>(e)) {
2079     assert(!gse->isResultDependent());
2080 
2081     unsigned n = gse->getNumAssocs();
2082     SmallVector<Expr*, 4> subExprs(n);
2083     SmallVector<TypeSourceInfo*, 4> subTypes(n);
2084     for (unsigned i = 0; i != n; ++i) {
2085       subTypes[i] = gse->getAssocTypeSourceInfo(i);
2086       Expr *sub = gse->getAssocExpr(i);
2087       if (i == gse->getResultIndex())
2088         sub = stripARCUnbridgedCast(sub);
2089       subExprs[i] = sub;
2090     }
2091 
2092     return new (Context) GenericSelectionExpr(Context, gse->getGenericLoc(),
2093                                               gse->getControllingExpr(),
2094                                               subTypes.data(), subExprs.data(),
2095                                               n, gse->getDefaultLoc(),
2096                                               gse->getRParenLoc(),
2097                                        gse->containsUnexpandedParameterPack(),
2098                                               gse->getResultIndex());
2099   } else {
2100     assert(isa<ImplicitCastExpr>(e) && "bad form of unbridged cast!");
2101     return cast<ImplicitCastExpr>(e)->getSubExpr();
2102   }
2103 }
2104 
2105 bool Sema::CheckObjCARCUnavailableWeakConversion(QualType castType,
2106                                                  QualType exprType) {
2107   QualType canCastType =
2108     Context.getCanonicalType(castType).getUnqualifiedType();
2109   QualType canExprType =
2110     Context.getCanonicalType(exprType).getUnqualifiedType();
2111   if (isa<ObjCObjectPointerType>(canCastType) &&
2112       castType.getObjCLifetime() == Qualifiers::OCL_Weak &&
2113       canExprType->isObjCObjectPointerType()) {
2114     if (const ObjCObjectPointerType *ObjT =
2115         canExprType->getAs<ObjCObjectPointerType>())
2116       if (ObjT->getInterfaceDecl()->isArcWeakrefUnavailable())
2117         return false;
2118   }
2119   return true;
2120 }
2121 
2122 /// Look for an ObjCReclaimReturnedObject cast and destroy it.
2123 static Expr *maybeUndoReclaimObject(Expr *e) {
2124   // For now, we just undo operands that are *immediately* reclaim
2125   // expressions, which prevents the vast majority of potential
2126   // problems here.  To catch them all, we'd need to rebuild arbitrary
2127   // value-propagating subexpressions --- we can't reliably rebuild
2128   // in-place because of expression sharing.
2129   if (ImplicitCastExpr *ice = dyn_cast<ImplicitCastExpr>(e))
2130     if (ice->getCastKind() == CK_ARCReclaimReturnedObject)
2131       return ice->getSubExpr();
2132 
2133   return e;
2134 }
2135 
2136 ExprResult Sema::BuildObjCBridgedCast(SourceLocation LParenLoc,
2137                                       ObjCBridgeCastKind Kind,
2138                                       SourceLocation BridgeKeywordLoc,
2139                                       TypeSourceInfo *TSInfo,
2140                                       Expr *SubExpr) {
2141   ExprResult SubResult = UsualUnaryConversions(SubExpr);
2142   if (SubResult.isInvalid()) return ExprError();
2143   SubExpr = SubResult.take();
2144 
2145   QualType T = TSInfo->getType();
2146   QualType FromType = SubExpr->getType();
2147 
2148   CastKind CK;
2149 
2150   bool MustConsume = false;
2151   if (T->isDependentType() || SubExpr->isTypeDependent()) {
2152     // Okay: we'll build a dependent expression type.
2153     CK = CK_Dependent;
2154   } else if (T->isObjCARCBridgableType() && FromType->isCARCBridgableType()) {
2155     // Casting CF -> id
2156     CK = (T->isBlockPointerType() ? CK_AnyPointerToBlockPointerCast
2157                                   : CK_CPointerToObjCPointerCast);
2158     switch (Kind) {
2159     case OBC_Bridge:
2160       break;
2161 
2162     case OBC_BridgeRetained:
2163       Diag(BridgeKeywordLoc, diag::err_arc_bridge_cast_wrong_kind)
2164         << 2
2165         << FromType
2166         << (T->isBlockPointerType()? 1 : 0)
2167         << T
2168         << SubExpr->getSourceRange()
2169         << Kind;
2170       Diag(BridgeKeywordLoc, diag::note_arc_bridge)
2171         << FixItHint::CreateReplacement(BridgeKeywordLoc, "__bridge");
2172       Diag(BridgeKeywordLoc, diag::note_arc_bridge_transfer)
2173         << FromType
2174         << FixItHint::CreateReplacement(BridgeKeywordLoc,
2175                                         "__bridge_transfer ");
2176 
2177       Kind = OBC_Bridge;
2178       break;
2179 
2180     case OBC_BridgeTransfer:
2181       // We must consume the Objective-C object produced by the cast.
2182       MustConsume = true;
2183       break;
2184     }
2185   } else if (T->isCARCBridgableType() && FromType->isObjCARCBridgableType()) {
2186     // Okay: id -> CF
2187     CK = CK_BitCast;
2188     switch (Kind) {
2189     case OBC_Bridge:
2190       // Reclaiming a value that's going to be __bridge-casted to CF
2191       // is very dangerous, so we don't do it.
2192       SubExpr = maybeUndoReclaimObject(SubExpr);
2193       break;
2194 
2195     case OBC_BridgeRetained:
2196       // Produce the object before casting it.
2197       SubExpr = ImplicitCastExpr::Create(Context, FromType,
2198                                          CK_ARCProduceObject,
2199                                          SubExpr, 0, VK_RValue);
2200       break;
2201 
2202     case OBC_BridgeTransfer:
2203       Diag(BridgeKeywordLoc, diag::err_arc_bridge_cast_wrong_kind)
2204         << (FromType->isBlockPointerType()? 1 : 0)
2205         << FromType
2206         << 2
2207         << T
2208         << SubExpr->getSourceRange()
2209         << Kind;
2210 
2211       Diag(BridgeKeywordLoc, diag::note_arc_bridge)
2212         << FixItHint::CreateReplacement(BridgeKeywordLoc, "__bridge ");
2213       Diag(BridgeKeywordLoc, diag::note_arc_bridge_retained)
2214         << T
2215         << FixItHint::CreateReplacement(BridgeKeywordLoc, "__bridge_retained ");
2216 
2217       Kind = OBC_Bridge;
2218       break;
2219     }
2220   } else {
2221     Diag(LParenLoc, diag::err_arc_bridge_cast_incompatible)
2222       << FromType << T << Kind
2223       << SubExpr->getSourceRange()
2224       << TSInfo->getTypeLoc().getSourceRange();
2225     return ExprError();
2226   }
2227 
2228   Expr *Result = new (Context) ObjCBridgedCastExpr(LParenLoc, Kind, CK,
2229                                                    BridgeKeywordLoc,
2230                                                    TSInfo, SubExpr);
2231 
2232   if (MustConsume) {
2233     ExprNeedsCleanups = true;
2234     Result = ImplicitCastExpr::Create(Context, T, CK_ARCConsumeObject, Result,
2235                                       0, VK_RValue);
2236   }
2237 
2238   return Result;
2239 }
2240 
2241 ExprResult Sema::ActOnObjCBridgedCast(Scope *S,
2242                                       SourceLocation LParenLoc,
2243                                       ObjCBridgeCastKind Kind,
2244                                       SourceLocation BridgeKeywordLoc,
2245                                       ParsedType Type,
2246                                       SourceLocation RParenLoc,
2247                                       Expr *SubExpr) {
2248   TypeSourceInfo *TSInfo = 0;
2249   QualType T = GetTypeFromParser(Type, &TSInfo);
2250   if (!TSInfo)
2251     TSInfo = Context.getTrivialTypeSourceInfo(T, LParenLoc);
2252   return BuildObjCBridgedCast(LParenLoc, Kind, BridgeKeywordLoc, TSInfo,
2253                               SubExpr);
2254 }
2255