Tune the lookup logic in Sema::ActOnInstanceMessage() to handle private methods (declared within the implementation).
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@44041 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/Sema/SemaExpr.cpp b/Sema/SemaExpr.cpp
index 976069e..87b3d08 100644
--- a/Sema/SemaExpr.cpp
+++ b/Sema/SemaExpr.cpp
@@ -2139,6 +2139,23 @@
if (receiverType == Context.getObjcIdType() ||
receiverType == Context.getObjcClassType()) {
Method = InstanceMethodPool[Sel].Method;
+ // If we didn't find an public method, look for a private one.
+ if (!Method && CurMethodDecl) {
+ NamedDecl *impCxt = CurMethodDecl->getMethodContext();
+ if (ObjcImplementationDecl *IMD =
+ dyn_cast<ObjcImplementationDecl>(impCxt)) {
+ if (receiverType == Context.getObjcIdType())
+ Method = IMD->lookupInstanceMethod(Sel);
+ else
+ Method = IMD->lookupClassMethod(Sel);
+ } else if (ObjcCategoryImplDecl *CID =
+ dyn_cast<ObjcCategoryImplDecl>(impCxt)) {
+ if (receiverType == Context.getObjcIdType())
+ Method = CID->lookupInstanceMethod(Sel);
+ else
+ Method = CID->lookupClassMethod(Sel);
+ }
+ }
if (!Method) {
Diag(lbrac, diag::warn_method_not_found, std::string("-"), Sel.getName(),
SourceRange(lbrac, rbrac));