SemaExprObjC.cpp revision d9fd7647e286723d100db4cfeab31ec022eec629
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 "Sema.h"
15#include "clang/AST/ASTContext.h"
16#include "clang/AST/DeclObjC.h"
17#include "clang/AST/ExprObjC.h"
18#include "llvm/ADT/SmallString.h"
19#include "clang/Lex/Preprocessor.h"
20
21using namespace clang;
22
23Sema::ExprResult Sema::ParseObjCStringLiteral(SourceLocation *AtLocs,
24                                              ExprTy **strings,
25                                              unsigned NumStrings) {
26  StringLiteral **Strings = reinterpret_cast<StringLiteral**>(strings);
27
28  // Most ObjC strings are formed out of a single piece.  However, we *can*
29  // have strings formed out of multiple @ strings with multiple pptokens in
30  // each one, e.g. @"foo" "bar" @"baz" "qux"   which need to be turned into one
31  // StringLiteral for ObjCStringLiteral to hold onto.
32  StringLiteral *S = Strings[0];
33
34  // If we have a multi-part string, merge it all together.
35  if (NumStrings != 1) {
36    // Concatenate objc strings.
37    llvm::SmallString<128> StrBuf;
38    llvm::SmallVector<SourceLocation, 8> StrLocs;
39
40    for (unsigned i = 0; i != NumStrings; ++i) {
41      S = Strings[i];
42
43      // ObjC strings can't be wide.
44      if (S->isWide()) {
45        Diag(S->getLocStart(), diag::err_cfstring_literal_not_string_constant)
46          << S->getSourceRange();
47        return true;
48      }
49
50      // Get the string data.
51      StrBuf.append(S->getStrData(), S->getStrData()+S->getByteLength());
52
53      // Get the locations of the string tokens.
54      StrLocs.append(S->tokloc_begin(), S->tokloc_end());
55
56      // Free the temporary string.
57      S->Destroy(Context);
58    }
59
60    // Create the aggregate string with the appropriate content and location
61    // information.
62    S = StringLiteral::Create(Context, &StrBuf[0], StrBuf.size(), false,
63                              Context.getPointerType(Context.CharTy),
64                              &StrLocs[0], StrLocs.size());
65  }
66
67  // Verify that this composite string is acceptable for ObjC strings.
68  if (CheckObjCString(S))
69    return true;
70
71  // Initialize the constant string interface lazily. This assumes
72  // the NSString interface is seen in this translation unit. Note: We
73  // don't use NSConstantString, since the runtime team considers this
74  // interface private (even though it appears in the header files).
75  QualType Ty = Context.getObjCConstantStringInterface();
76  if (!Ty.isNull()) {
77    Ty = Context.getPointerType(Ty);
78  } else {
79    IdentifierInfo *NSIdent = &Context.Idents.get("NSString");
80    NamedDecl *IF = LookupName(TUScope, NSIdent, LookupOrdinaryName);
81    if (ObjCInterfaceDecl *StrIF = dyn_cast_or_null<ObjCInterfaceDecl>(IF)) {
82      Context.setObjCConstantStringInterface(StrIF);
83      Ty = Context.getObjCConstantStringInterface();
84      Ty = Context.getPointerType(Ty);
85    } else {
86      // If there is no NSString interface defined then treat constant
87      // strings as untyped objects and let the runtime figure it out later.
88      Ty = Context.getObjCIdType();
89    }
90  }
91
92  return new (Context) ObjCStringLiteral(S, Ty, AtLocs[0]);
93}
94
95Sema::ExprResult Sema::ParseObjCEncodeExpression(SourceLocation AtLoc,
96                                                 SourceLocation EncodeLoc,
97                                                 SourceLocation LParenLoc,
98                                                 TypeTy *ty,
99                                                 SourceLocation RParenLoc) {
100  QualType EncodedType = QualType::getFromOpaquePtr(ty);
101
102  std::string Str;
103  Context.getObjCEncodingForType(EncodedType, Str);
104
105  // The type of @encode is the same as the type of the corresponding string,
106  // which is an array type.
107  QualType StrTy = Context.CharTy;
108  // A C++ string literal has a const-qualified element type (C++ 2.13.4p1).
109  if (getLangOptions().CPlusPlus)
110    StrTy.addConst();
111  StrTy = Context.getConstantArrayType(StrTy, llvm::APInt(32, Str.size()+1),
112                                       ArrayType::Normal, 0);
113
114  return new (Context) ObjCEncodeExpr(StrTy, EncodedType, AtLoc, RParenLoc);
115}
116
117Sema::ExprResult Sema::ParseObjCSelectorExpression(Selector Sel,
118                                                   SourceLocation AtLoc,
119                                                   SourceLocation SelLoc,
120                                                   SourceLocation LParenLoc,
121                                                   SourceLocation RParenLoc) {
122  QualType Ty = Context.getObjCSelType();
123  return new (Context) ObjCSelectorExpr(Ty, Sel, AtLoc, RParenLoc);
124}
125
126Sema::ExprResult Sema::ParseObjCProtocolExpression(IdentifierInfo *ProtocolId,
127                                                   SourceLocation AtLoc,
128                                                   SourceLocation ProtoLoc,
129                                                   SourceLocation LParenLoc,
130                                                   SourceLocation RParenLoc) {
131  ObjCProtocolDecl* PDecl = ObjCProtocols[ProtocolId];
132  if (!PDecl) {
133    Diag(ProtoLoc, diag::err_undeclared_protocol) << ProtocolId;
134    return true;
135  }
136
137  QualType Ty = Context.getObjCProtoType();
138  if (Ty.isNull())
139    return true;
140  Ty = Context.getPointerType(Ty);
141  return new (Context) ObjCProtocolExpr(Ty, PDecl, AtLoc, RParenLoc);
142}
143
144bool Sema::CheckMessageArgumentTypes(Expr **Args, unsigned NumArgs,
145                                     Selector Sel, ObjCMethodDecl *Method,
146                                     bool isClassMessage,
147                                     SourceLocation lbrac, SourceLocation rbrac,
148                                     QualType &ReturnType) {
149  if (!Method) {
150    // Apply default argument promotion as for (C99 6.5.2.2p6).
151    for (unsigned i = 0; i != NumArgs; i++)
152      DefaultArgumentPromotion(Args[i]);
153
154    unsigned DiagID = isClassMessage ? diag::warn_class_method_not_found :
155                                       diag::warn_inst_method_not_found;
156    Diag(lbrac, DiagID)
157      << Sel << isClassMessage << SourceRange(lbrac, rbrac);
158    ReturnType = Context.getObjCIdType();
159    return false;
160  }
161
162  ReturnType = Method->getResultType();
163
164  unsigned NumNamedArgs = Sel.getNumArgs();
165  assert(NumArgs >= NumNamedArgs && "Too few arguments for selector!");
166
167  bool anyIncompatibleArgs = false;
168  for (unsigned i = 0; i < NumNamedArgs; i++) {
169    Expr *argExpr = Args[i];
170    assert(argExpr && "CheckMessageArgumentTypes(): missing expression");
171
172    QualType lhsType = Method->param_begin()[i]->getType();
173    QualType rhsType = argExpr->getType();
174
175    // If necessary, apply function/array conversion. C99 6.7.5.3p[7,8].
176    if (lhsType->isArrayType())
177      lhsType = Context.getArrayDecayedType(lhsType);
178    else if (lhsType->isFunctionType())
179      lhsType = Context.getPointerType(lhsType);
180
181    AssignConvertType Result =
182      CheckSingleAssignmentConstraints(lhsType, argExpr);
183    if (Args[i] != argExpr) // The expression was converted.
184      Args[i] = argExpr; // Make sure we store the converted expression.
185
186    anyIncompatibleArgs |=
187      DiagnoseAssignmentResult(Result, argExpr->getLocStart(), lhsType, rhsType,
188                               argExpr, "sending");
189  }
190
191  // Promote additional arguments to variadic methods.
192  if (Method->isVariadic()) {
193    for (unsigned i = NumNamedArgs; i < NumArgs; ++i)
194      DefaultVariadicArgumentPromotion(Args[i], VariadicMethod);
195  } else {
196    // Check for extra arguments to non-variadic methods.
197    if (NumArgs != NumNamedArgs) {
198      Diag(Args[NumNamedArgs]->getLocStart(),
199           diag::err_typecheck_call_too_many_args)
200        << 2 /*method*/ << Method->getSourceRange()
201        << SourceRange(Args[NumNamedArgs]->getLocStart(),
202                       Args[NumArgs-1]->getLocEnd());
203    }
204  }
205
206  return anyIncompatibleArgs;
207}
208
209bool Sema::isSelfExpr(Expr *RExpr) {
210  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(RExpr))
211    if (DRE->getDecl()->getIdentifier() == &Context.Idents.get("self"))
212      return true;
213  return false;
214}
215
216// Helper method for ActOnClassMethod/ActOnInstanceMethod.
217// Will search "local" class/category implementations for a method decl.
218// If failed, then we search in class's root for an instance method.
219// Returns 0 if no method is found.
220ObjCMethodDecl *Sema::LookupPrivateClassMethod(Selector Sel,
221                                          ObjCInterfaceDecl *ClassDecl) {
222  ObjCMethodDecl *Method = 0;
223  // lookup in class and all superclasses
224  while (ClassDecl && !Method) {
225    if (ObjCImplementationDecl *ImpDecl =
226        ObjCImplementations[ClassDecl->getIdentifier()])
227      Method = ImpDecl->getClassMethod(Sel);
228
229    // Look through local category implementations associated with the class.
230    if (!Method) {
231      for (unsigned i = 0; i < ObjCCategoryImpls.size() && !Method; i++) {
232        if (ObjCCategoryImpls[i]->getClassInterface() == ClassDecl)
233          Method = ObjCCategoryImpls[i]->getClassMethod(Sel);
234      }
235    }
236
237    // Before we give up, check if the selector is an instance method.
238    // But only in the root. This matches gcc's behaviour and what the
239    // runtime expects.
240    if (!Method && !ClassDecl->getSuperClass()) {
241      Method = ClassDecl->lookupInstanceMethod(Sel);
242      // Look through local category implementations associated
243      // with the root class.
244      if (!Method)
245        Method = LookupPrivateInstanceMethod(Sel, ClassDecl);
246    }
247
248    ClassDecl = ClassDecl->getSuperClass();
249  }
250  return Method;
251}
252
253ObjCMethodDecl *Sema::LookupPrivateInstanceMethod(Selector Sel,
254                                              ObjCInterfaceDecl *ClassDecl) {
255  ObjCMethodDecl *Method = 0;
256  while (ClassDecl && !Method) {
257    // If we have implementations in scope, check "private" methods.
258    if (ObjCImplementationDecl *ImpDecl =
259        ObjCImplementations[ClassDecl->getIdentifier()])
260      Method = ImpDecl->getInstanceMethod(Sel);
261
262    // Look through local category implementations associated with the class.
263    if (!Method) {
264      for (unsigned i = 0; i < ObjCCategoryImpls.size() && !Method; i++) {
265        if (ObjCCategoryImpls[i]->getClassInterface() == ClassDecl)
266          Method = ObjCCategoryImpls[i]->getInstanceMethod(Sel);
267      }
268    }
269    ClassDecl = ClassDecl->getSuperClass();
270  }
271  return Method;
272}
273
274Action::OwningExprResult Sema::ActOnClassPropertyRefExpr(
275  IdentifierInfo &receiverName,
276  IdentifierInfo &propertyName,
277  SourceLocation &receiverNameLoc,
278  SourceLocation &propertyNameLoc) {
279
280  ObjCInterfaceDecl *IFace = getObjCInterfaceDecl(&receiverName);
281
282  // Search for a declared property first.
283
284  Selector Sel = PP.getSelectorTable().getNullarySelector(&propertyName);
285  ObjCMethodDecl *Getter = IFace->lookupClassMethod(Sel);
286
287  // If this reference is in an @implementation, check for 'private' methods.
288  if (!Getter)
289    if (ObjCMethodDecl *CurMeth = getCurMethodDecl())
290      if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface())
291        if (ObjCImplementationDecl *ImpDecl =
292            ObjCImplementations[ClassDecl->getIdentifier()])
293          Getter = ImpDecl->getClassMethod(Sel);
294
295  if (Getter) {
296    // FIXME: refactor/share with ActOnMemberReference().
297    // Check if we can reference this property.
298    if (DiagnoseUseOfDecl(Getter, propertyNameLoc))
299      return ExprError();
300  }
301
302  // Look for the matching setter, in case it is needed.
303  Selector SetterSel =
304    SelectorTable::constructSetterName(PP.getIdentifierTable(),
305                                       PP.getSelectorTable(), &propertyName);
306
307  ObjCMethodDecl *Setter = IFace->lookupClassMethod(SetterSel);
308  if (!Setter) {
309    // If this reference is in an @implementation, also check for 'private'
310    // methods.
311    if (ObjCMethodDecl *CurMeth = getCurMethodDecl())
312      if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface())
313        if (ObjCImplementationDecl *ImpDecl =
314              ObjCImplementations[ClassDecl->getIdentifier()])
315          Setter = ImpDecl->getClassMethod(SetterSel);
316  }
317  // Look through local category implementations associated with the class.
318  if (!Setter) {
319    for (unsigned i = 0; i < ObjCCategoryImpls.size() && !Setter; i++) {
320      if (ObjCCategoryImpls[i]->getClassInterface() == IFace)
321        Setter = ObjCCategoryImpls[i]->getClassMethod(SetterSel);
322    }
323  }
324
325  if (Setter && DiagnoseUseOfDecl(Setter, propertyNameLoc))
326    return ExprError();
327
328  if (Getter || Setter) {
329    QualType PType;
330
331    if (Getter)
332      PType = Getter->getResultType();
333    else {
334      for (ObjCMethodDecl::param_iterator PI = Setter->param_begin(),
335           E = Setter->param_end(); PI != E; ++PI)
336        PType = (*PI)->getType();
337    }
338    return Owned(new (Context) ObjCKVCRefExpr(Getter, PType, Setter,
339                                  propertyNameLoc, IFace, receiverNameLoc));
340  }
341  return ExprError(Diag(propertyNameLoc, diag::err_property_not_found)
342                     << &propertyName << Context.getObjCInterfaceType(IFace));
343}
344
345
346// ActOnClassMessage - used for both unary and keyword messages.
347// ArgExprs is optional - if it is present, the number of expressions
348// is obtained from Sel.getNumArgs().
349Sema::ExprResult Sema::ActOnClassMessage(
350  Scope *S,
351  IdentifierInfo *receiverName, Selector Sel,
352  SourceLocation lbrac, SourceLocation receiverLoc,
353  SourceLocation selectorLoc, SourceLocation rbrac,
354  ExprTy **Args, unsigned NumArgs)
355{
356  assert(receiverName && "missing receiver class name");
357
358  Expr **ArgExprs = reinterpret_cast<Expr **>(Args);
359  ObjCInterfaceDecl* ClassDecl = 0;
360  bool isSuper = false;
361
362  if (receiverName->isStr("super")) {
363    if (getCurMethodDecl()) {
364      isSuper = true;
365      ObjCInterfaceDecl *OID = getCurMethodDecl()->getClassInterface();
366      if (!OID)
367        return Diag(lbrac, diag::error_no_super_class_message)
368                      << getCurMethodDecl()->getDeclName();
369      ClassDecl = OID->getSuperClass();
370      if (!ClassDecl)
371        return Diag(lbrac, diag::error_no_super_class) << OID->getDeclName();
372      if (getCurMethodDecl()->isInstanceMethod()) {
373        QualType superTy = Context.getObjCInterfaceType(ClassDecl);
374        superTy = Context.getPointerType(superTy);
375        ExprResult ReceiverExpr = new (Context) ObjCSuperExpr(SourceLocation(),
376                                                              superTy);
377        // We are really in an instance method, redirect.
378        return ActOnInstanceMessage(ReceiverExpr.get(), Sel, lbrac,
379                                    selectorLoc, rbrac, Args, NumArgs);
380      }
381      // We are sending a message to 'super' within a class method. Do nothing,
382      // the receiver will pass through as 'super' (how convenient:-).
383    } else {
384      // 'super' has been used outside a method context. If a variable named
385      // 'super' has been declared, redirect. If not, produce a diagnostic.
386      NamedDecl *SuperDecl = LookupName(S, receiverName, LookupOrdinaryName);
387      ValueDecl *VD = dyn_cast_or_null<ValueDecl>(SuperDecl);
388      if (VD) {
389        ExprResult ReceiverExpr = new (Context) DeclRefExpr(VD, VD->getType(),
390                                                            receiverLoc);
391        // We are really in an instance method, redirect.
392        return ActOnInstanceMessage(ReceiverExpr.get(), Sel, lbrac,
393                                    selectorLoc, rbrac, Args, NumArgs);
394      }
395      return Diag(receiverLoc, diag::err_undeclared_var_use) << receiverName;
396    }
397  } else
398    ClassDecl = getObjCInterfaceDecl(receiverName);
399
400  // The following code allows for the following GCC-ism:
401  //
402  //  typedef XCElementDisplayRect XCElementGraphicsRect;
403  //
404  //  @implementation XCRASlice
405  //  - whatever { // Note that XCElementGraphicsRect is a typedef name.
406  //    _sGraphicsDelegate =[[XCElementGraphicsRect alloc] init];
407  //  }
408  //
409  // If necessary, the following lookup could move to getObjCInterfaceDecl().
410  if (!ClassDecl) {
411    NamedDecl *IDecl = LookupName(TUScope, receiverName, LookupOrdinaryName);
412    if (TypedefDecl *OCTD = dyn_cast_or_null<TypedefDecl>(IDecl)) {
413      const ObjCInterfaceType *OCIT;
414      OCIT = OCTD->getUnderlyingType()->getAsObjCInterfaceType();
415      if (!OCIT) {
416        Diag(receiverLoc, diag::err_invalid_receiver_to_message);
417        return true;
418      }
419      ClassDecl = OCIT->getDecl();
420    }
421  }
422  assert(ClassDecl && "missing interface declaration");
423  ObjCMethodDecl *Method = 0;
424  QualType returnType;
425  Method = ClassDecl->lookupClassMethod(Sel);
426
427  // If we have an implementation in scope, check "private" methods.
428  if (!Method)
429    Method = LookupPrivateClassMethod(Sel, ClassDecl);
430
431  if (Method && DiagnoseUseOfDecl(Method, receiverLoc))
432    return true;
433
434  if (CheckMessageArgumentTypes(ArgExprs, NumArgs, Sel, Method, true,
435                                lbrac, rbrac, returnType))
436    return true;
437
438  // If we have the ObjCInterfaceDecl* for the class that is receiving
439  // the message, use that to construct the ObjCMessageExpr.  Otherwise
440  // pass on the IdentifierInfo* for the class.
441  // FIXME: need to do a better job handling 'super' usage within a class
442  // For now, we simply pass the "super" identifier through (which isn't
443  // consistent with instance methods.
444  if (isSuper)
445    return new (Context) ObjCMessageExpr(receiverName, Sel, returnType, Method,
446                                         lbrac, rbrac, ArgExprs, NumArgs);
447  else
448    return new (Context) ObjCMessageExpr(ClassDecl, Sel, returnType, Method,
449                                         lbrac, rbrac, ArgExprs, NumArgs);
450}
451
452// ActOnInstanceMessage - used for both unary and keyword messages.
453// ArgExprs is optional - if it is present, the number of expressions
454// is obtained from Sel.getNumArgs().
455Sema::ExprResult Sema::ActOnInstanceMessage(ExprTy *receiver, Selector Sel,
456                                            SourceLocation lbrac,
457                                            SourceLocation receiverLoc,
458                                            SourceLocation rbrac,
459                                            ExprTy **Args, unsigned NumArgs) {
460  assert(receiver && "missing receiver expression");
461
462  Expr **ArgExprs = reinterpret_cast<Expr **>(Args);
463  Expr *RExpr = static_cast<Expr *>(receiver);
464  QualType returnType;
465
466  QualType ReceiverCType =
467    Context.getCanonicalType(RExpr->getType()).getUnqualifiedType();
468
469  // Handle messages to 'super'.
470  if (isa<ObjCSuperExpr>(RExpr)) {
471    ObjCMethodDecl *Method = 0;
472    if (ObjCMethodDecl *CurMeth = getCurMethodDecl()) {
473      // If we have an interface in scope, check 'super' methods.
474      if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface())
475        if (ObjCInterfaceDecl *SuperDecl = ClassDecl->getSuperClass()) {
476          Method = SuperDecl->lookupInstanceMethod(Sel);
477
478          if (!Method)
479            // If we have implementations in scope, check "private" methods.
480            Method = LookupPrivateInstanceMethod(Sel, SuperDecl);
481        }
482    }
483
484    if (Method && DiagnoseUseOfDecl(Method, receiverLoc))
485      return true;
486
487    if (CheckMessageArgumentTypes(ArgExprs, NumArgs, Sel, Method, false,
488                                  lbrac, rbrac, returnType))
489      return true;
490    return new (Context) ObjCMessageExpr(RExpr, Sel, returnType, Method, lbrac,
491                                         rbrac, ArgExprs, NumArgs);
492  }
493
494  // Handle messages to id.
495  if (ReceiverCType == Context.getCanonicalType(Context.getObjCIdType()) ||
496      ReceiverCType->isBlockPointerType()) {
497    ObjCMethodDecl *Method = LookupInstanceMethodInGlobalPool(
498                               Sel, SourceRange(lbrac,rbrac));
499    if (!Method)
500      Method = FactoryMethodPool[Sel].Method;
501    if (CheckMessageArgumentTypes(ArgExprs, NumArgs, Sel, Method, false,
502                                  lbrac, rbrac, returnType))
503      return true;
504    return new (Context) ObjCMessageExpr(RExpr, Sel, returnType, Method, lbrac,
505                                         rbrac, ArgExprs, NumArgs);
506  }
507
508  // Handle messages to Class.
509  if (ReceiverCType == Context.getCanonicalType(Context.getObjCClassType())) {
510    ObjCMethodDecl *Method = 0;
511
512    if (ObjCMethodDecl *CurMeth = getCurMethodDecl()) {
513      if (ObjCInterfaceDecl *ClassDecl = CurMeth->getClassInterface()) {
514        // First check the public methods in the class interface.
515        Method = ClassDecl->lookupClassMethod(Sel);
516
517        if (!Method)
518          Method = LookupPrivateClassMethod(Sel, ClassDecl);
519      }
520      if (Method && DiagnoseUseOfDecl(Method, receiverLoc))
521        return true;
522    }
523    if (!Method) {
524      // If not messaging 'self', look for any factory method named 'Sel'.
525      if (!isSelfExpr(RExpr)) {
526        Method = FactoryMethodPool[Sel].Method;
527        if (!Method) {
528          Method = LookupInstanceMethodInGlobalPool(
529                                   Sel, SourceRange(lbrac,rbrac));
530        }
531      }
532    }
533    if (CheckMessageArgumentTypes(ArgExprs, NumArgs, Sel, Method, false,
534                                  lbrac, rbrac, returnType))
535      return true;
536    return new (Context) ObjCMessageExpr(RExpr, Sel, returnType, Method, lbrac,
537                                         rbrac, ArgExprs, NumArgs);
538  }
539
540  ObjCMethodDecl *Method = 0;
541  ObjCInterfaceDecl* ClassDecl = 0;
542
543  // We allow sending a message to a qualified ID ("id<foo>"), which is ok as
544  // long as one of the protocols implements the selector (if not, warn).
545  if (ObjCQualifiedIdType *QIT = dyn_cast<ObjCQualifiedIdType>(ReceiverCType)) {
546    // Search protocols for instance methods.
547    for (unsigned i = 0; i < QIT->getNumProtocols(); i++) {
548      ObjCProtocolDecl *PDecl = QIT->getProtocols(i);
549      if (PDecl && (Method = PDecl->lookupInstanceMethod(Sel)))
550        break;
551    }
552  } else if (const ObjCInterfaceType *OCIType =
553                ReceiverCType->getAsPointerToObjCInterfaceType()) {
554    // We allow sending a message to a pointer to an interface (an object).
555
556    ClassDecl = OCIType->getDecl();
557    // FIXME: consider using LookupInstanceMethodInGlobalPool, since it will be
558    // faster than the following method (which can do *many* linear searches).
559    // The idea is to add class info to InstanceMethodPool.
560    Method = ClassDecl->lookupInstanceMethod(Sel);
561
562    if (!Method) {
563      // Search protocol qualifiers.
564      for (ObjCQualifiedInterfaceType::qual_iterator QI = OCIType->qual_begin(),
565           E = OCIType->qual_end(); QI != E; ++QI) {
566        if ((Method = (*QI)->lookupInstanceMethod(Sel)))
567          break;
568      }
569    }
570    if (!Method) {
571      // If we have implementations in scope, check "private" methods.
572      Method = LookupPrivateInstanceMethod(Sel, ClassDecl);
573
574      if (!Method && !isSelfExpr(RExpr)) {
575        // If we still haven't found a method, look in the global pool. This
576        // behavior isn't very desirable, however we need it for GCC
577        // compatibility. FIXME: should we deviate??
578        if (OCIType->qual_empty()) {
579          Method = LookupInstanceMethodInGlobalPool(
580                               Sel, SourceRange(lbrac,rbrac));
581          if (Method && !OCIType->getDecl()->isForwardDecl())
582            Diag(lbrac, diag::warn_maynot_respond)
583              << OCIType->getDecl()->getIdentifier()->getName() << Sel;
584        }
585      }
586    }
587    if (Method && DiagnoseUseOfDecl(Method, receiverLoc))
588      return true;
589  } else if (!Context.getObjCIdType().isNull() &&
590             (ReceiverCType->isPointerType() ||
591              (ReceiverCType->isIntegerType() &&
592               ReceiverCType->isScalarType()))) {
593    // Implicitly convert integers and pointers to 'id' but emit a warning.
594    Diag(lbrac, diag::warn_bad_receiver_type)
595      << RExpr->getType() << RExpr->getSourceRange();
596    ImpCastExprToType(RExpr, Context.getObjCIdType());
597  } else {
598    // Reject other random receiver types (e.g. structs).
599    Diag(lbrac, diag::err_bad_receiver_type)
600      << RExpr->getType() << RExpr->getSourceRange();
601    return true;
602  }
603
604  if (CheckMessageArgumentTypes(ArgExprs, NumArgs, Sel, Method, false,
605                                lbrac, rbrac, returnType))
606    return true;
607  return new (Context) ObjCMessageExpr(RExpr, Sel, returnType, Method, lbrac,
608                                       rbrac, ArgExprs, NumArgs);
609}
610
611//===----------------------------------------------------------------------===//
612// ObjCQualifiedIdTypesAreCompatible - Compatibility testing for qualified id's.
613//===----------------------------------------------------------------------===//
614
615/// ProtocolCompatibleWithProtocol - return 'true' if 'lProto' is in the
616/// inheritance hierarchy of 'rProto'.
617static bool ProtocolCompatibleWithProtocol(ObjCProtocolDecl *lProto,
618                                           ObjCProtocolDecl *rProto) {
619  if (lProto == rProto)
620    return true;
621  for (ObjCProtocolDecl::protocol_iterator PI = rProto->protocol_begin(),
622       E = rProto->protocol_end(); PI != E; ++PI)
623    if (ProtocolCompatibleWithProtocol(lProto, *PI))
624      return true;
625  return false;
626}
627
628/// ClassImplementsProtocol - Checks that 'lProto' protocol
629/// has been implemented in IDecl class, its super class or categories (if
630/// lookupCategory is true).
631static bool ClassImplementsProtocol(ObjCProtocolDecl *lProto,
632                                    ObjCInterfaceDecl *IDecl,
633                                    bool lookupCategory,
634                                    bool RHSIsQualifiedID = false) {
635
636  // 1st, look up the class.
637  const ObjCList<ObjCProtocolDecl> &Protocols =
638    IDecl->getReferencedProtocols();
639
640  for (ObjCList<ObjCProtocolDecl>::iterator PI = Protocols.begin(),
641       E = Protocols.end(); PI != E; ++PI) {
642    if (ProtocolCompatibleWithProtocol(lProto, *PI))
643      return true;
644    // This is dubious and is added to be compatible with gcc.
645    // In gcc, it is also allowed assigning a protocol-qualified 'id'
646    // type to a LHS object when protocol in qualified LHS is in list
647    // of protocols in the rhs 'id' object. This IMO, should be a bug.
648    // FIXME: Treat this as an extension, and flag this as an error when
649    //  GCC extensions are not enabled.
650    if (RHSIsQualifiedID && ProtocolCompatibleWithProtocol(*PI, lProto))
651      return true;
652  }
653
654  // 2nd, look up the category.
655  if (lookupCategory)
656    for (ObjCCategoryDecl *CDecl = IDecl->getCategoryList(); CDecl;
657         CDecl = CDecl->getNextClassCategory()) {
658      for (ObjCCategoryDecl::protocol_iterator PI = CDecl->protocol_begin(),
659           E = CDecl->protocol_end(); PI != E; ++PI)
660        if (ProtocolCompatibleWithProtocol(lProto, *PI))
661          return true;
662    }
663
664  // 3rd, look up the super class(s)
665  if (IDecl->getSuperClass())
666    return
667      ClassImplementsProtocol(lProto, IDecl->getSuperClass(), lookupCategory,
668                              RHSIsQualifiedID);
669
670  return false;
671}
672
673/// ObjCQualifiedIdTypesAreCompatible - We know that one of lhs/rhs is an
674/// ObjCQualifiedIDType.
675/// FIXME: Move to ASTContext::typesAreCompatible() and friends.
676bool Sema::ObjCQualifiedIdTypesAreCompatible(QualType lhs, QualType rhs,
677                                             bool compare) {
678  // Allow id<P..> and an 'id' or void* type in all cases.
679  if (const PointerType *PT = lhs->getAsPointerType()) {
680    QualType PointeeTy = PT->getPointeeType();
681    if (Context.isObjCIdStructType(PointeeTy) || PointeeTy->isVoidType())
682      return true;
683  } else if (const PointerType *PT = rhs->getAsPointerType()) {
684    QualType PointeeTy = PT->getPointeeType();
685    if (Context.isObjCIdStructType(PointeeTy) || PointeeTy->isVoidType())
686      return true;
687  }
688
689  if (const ObjCQualifiedIdType *lhsQID = lhs->getAsObjCQualifiedIdType()) {
690    const ObjCQualifiedIdType *rhsQID = rhs->getAsObjCQualifiedIdType();
691    const ObjCQualifiedInterfaceType *rhsQI = 0;
692    QualType rtype;
693
694    if (!rhsQID) {
695      // Not comparing two ObjCQualifiedIdType's?
696      if (!rhs->isPointerType()) return false;
697
698      rtype = rhs->getAsPointerType()->getPointeeType();
699      rhsQI = rtype->getAsObjCQualifiedInterfaceType();
700      if (rhsQI == 0) {
701        // If the RHS is a unqualified interface pointer "NSString*",
702        // make sure we check the class hierarchy.
703        if (const ObjCInterfaceType *IT = rtype->getAsObjCInterfaceType()) {
704          ObjCInterfaceDecl *rhsID = IT->getDecl();
705          for (unsigned i = 0; i != lhsQID->getNumProtocols(); ++i) {
706            // when comparing an id<P> on lhs with a static type on rhs,
707            // see if static class implements all of id's protocols, directly or
708            // through its super class and categories.
709            if (!ClassImplementsProtocol(lhsQID->getProtocols(i), rhsID, true))
710              return false;
711          }
712          return true;
713        }
714      }
715    }
716
717    ObjCQualifiedIdType::qual_iterator RHSProtoI, RHSProtoE;
718    if (rhsQI) { // We have a qualified interface (e.g. "NSObject<Proto> *").
719      RHSProtoI = rhsQI->qual_begin();
720      RHSProtoE = rhsQI->qual_end();
721    } else if (rhsQID) { // We have a qualified id (e.g. "id<Proto> *").
722      RHSProtoI = rhsQID->qual_begin();
723      RHSProtoE = rhsQID->qual_end();
724    } else {
725      return false;
726    }
727
728    for (unsigned i =0; i < lhsQID->getNumProtocols(); i++) {
729      ObjCProtocolDecl *lhsProto = lhsQID->getProtocols(i);
730      bool match = false;
731
732      // when comparing an id<P> on lhs with a static type on rhs,
733      // see if static class implements all of id's protocols, directly or
734      // through its super class and categories.
735      for (; RHSProtoI != RHSProtoE; ++RHSProtoI) {
736        ObjCProtocolDecl *rhsProto = *RHSProtoI;
737        if (ProtocolCompatibleWithProtocol(lhsProto, rhsProto) ||
738            (compare && ProtocolCompatibleWithProtocol(rhsProto, lhsProto))) {
739          match = true;
740          break;
741        }
742      }
743      if (rhsQI) {
744        // If the RHS is a qualified interface pointer "NSString<P>*",
745        // make sure we check the class hierarchy.
746        if (const ObjCInterfaceType *IT = rtype->getAsObjCInterfaceType()) {
747          ObjCInterfaceDecl *rhsID = IT->getDecl();
748          for (unsigned i = 0; i != lhsQID->getNumProtocols(); ++i) {
749            // when comparing an id<P> on lhs with a static type on rhs,
750            // see if static class implements all of id's protocols, directly or
751            // through its super class and categories.
752            if (ClassImplementsProtocol(lhsQID->getProtocols(i), rhsID, true)) {
753              match = true;
754              break;
755            }
756          }
757        }
758      }
759      if (!match)
760        return false;
761    }
762
763    return true;
764  }
765
766  const ObjCQualifiedIdType *rhsQID = rhs->getAsObjCQualifiedIdType();
767  assert(rhsQID && "One of the LHS/RHS should be id<x>");
768
769  if (!lhs->isPointerType())
770    return false;
771
772  QualType ltype = lhs->getAsPointerType()->getPointeeType();
773  if (const ObjCQualifiedInterfaceType *lhsQI =
774         ltype->getAsObjCQualifiedInterfaceType()) {
775    ObjCQualifiedIdType::qual_iterator LHSProtoI = lhsQI->qual_begin();
776    ObjCQualifiedIdType::qual_iterator LHSProtoE = lhsQI->qual_end();
777    for (; LHSProtoI != LHSProtoE; ++LHSProtoI) {
778      bool match = false;
779      ObjCProtocolDecl *lhsProto = *LHSProtoI;
780      for (unsigned j = 0; j < rhsQID->getNumProtocols(); j++) {
781        ObjCProtocolDecl *rhsProto = rhsQID->getProtocols(j);
782        if (ProtocolCompatibleWithProtocol(lhsProto, rhsProto) ||
783            (compare && ProtocolCompatibleWithProtocol(rhsProto, lhsProto))) {
784          match = true;
785          break;
786        }
787      }
788      if (!match)
789        return false;
790    }
791    return true;
792  }
793
794  if (const ObjCInterfaceType *IT = ltype->getAsObjCInterfaceType()) {
795    // for static type vs. qualified 'id' type, check that class implements
796    // all of 'id's protocols.
797    ObjCInterfaceDecl *lhsID = IT->getDecl();
798    for (unsigned j = 0; j < rhsQID->getNumProtocols(); j++) {
799      ObjCProtocolDecl *rhsProto = rhsQID->getProtocols(j);
800      if (!ClassImplementsProtocol(rhsProto, lhsID, compare, true))
801        return false;
802    }
803    return true;
804  }
805  return false;
806}
807
808