Objective-c: Conversion from type Class to any root class type is allowed

in class methods with no warning. //rdar://10109725


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@139818 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 30 insertions, 2 deletions

diff --git a/lib/Sema/SemaExpr.cpp b/lib/Sema/SemaExpr.cpp
index bb31507c23..35b38b101a 100644
--- a/lib/Sema/SemaExpr.cpp
+++ b/lib/Sema/SemaExpr.cpp
@@ -5227,6 +5227,35 @@ checkBlockPointerTypesForAssignment(Sema &S, QualType LHSType,
   return ConvTy;
 }
 
+static Sema::AssignConvertType
+checkObjCPointerTypesForAssignment(Sema &S, QualType LHSType,
+                                   QualType RHSType);
+/// checkClassTypes - Routine checks for conversion of "Class" type.
+// Conversion from type Class to any root class type in a class method
+// is allowed.
+static Sema::AssignConvertType
+checkClassTypes(Sema &S, QualType LHSType) {
+  // Conversion from type Class to any root class type is allowed.
+  DeclContext *DC = S.CurContext;
+  while (isa<BlockDecl>(DC))
+    DC = DC->getParent();
+  ObjCMethodDecl *MD = dyn_cast_or_null<ObjCMethodDecl>(DC);
+  if (MD && MD->isClassMethod()) {
+    ObjCInterfaceDecl *Root = 0;
+      if (ObjCInterfaceDecl * IDecl = MD->getClassInterface())
+        do
+          Root = IDecl;
+        while ((IDecl = IDecl->getSuperClass()));
+      if (Root){
+        QualType RHSType = 
+          S.Context.getObjCObjectPointerType(
+                                 S.Context.getObjCInterfaceType(Root));
+        return checkObjCPointerTypesForAssignment(S, LHSType, RHSType);
+      }
+  }
+  return Sema::IncompatiblePointer;
+}
+
 /// checkObjCPointerTypesForAssignment - Compares two objective-c pointer types
 /// for assignment compatibility.
 static Sema::AssignConvertType
@@ -5243,10 +5272,9 @@ checkObjCPointerTypesForAssignment(Sema &S, QualType LHSType,
     return Sema::Compatible;
   }
   if (RHSType->isObjCBuiltinType()) {
-    // Class is not compatible with ObjC object pointers.
     if (RHSType->isObjCClassType() && !LHSType->isObjCBuiltinType() &&
         !LHSType->isObjCQualifiedClassType())
-      return Sema::IncompatiblePointer;
+      return checkClassTypes(S, LHSType);
     return Sema::Compatible;
   }
   QualType lhptee = LHSType->getAs<ObjCObjectPointerType>()->getPointeeType();