commit c65b8bb647364ff1d8dfc1eec15ee2a26e7de726
author Chris Lattner <sabre@nondot.org> Sat Apr 11 19:27:54 2009 +0000
committer Chris Lattner <sabre@nondot.org> Sat Apr 11 19:27:54 2009 +0000
tree 56166a6206eb3173ef437c4a891d2ea80b56242a
parent fa0d9fe9806985194f9d7b2b7cbf1af5b57b438c

fix blocks to reject objc interfaces returned by value.  Also,
a block without a prototype should still coerce a return in it to
use the declared return type.


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

lib/Sema/SemaExpr.cpp

diff --git a/lib/Sema/SemaExpr.cpp b/lib/Sema/SemaExpr.cpp
index dda8654..f2820df 100644
--- a/lib/Sema/SemaExpr.cpp
+++ b/lib/Sema/SemaExpr.cpp
@@ -4655,11 +4655,18 @@
 
     CurBlock->hasPrototype = true;
     CurBlock->isVariadic = false;
-    Type *RetTy = T.getTypePtr()->getAsFunctionType()->getResultType()
-      .getTypePtr();
+    
+    QualType RetTy = T.getTypePtr()->getAsFunctionType()->getResultType();
+    
+    // Do not allow returning a objc interface by-value.
+    if (RetTy->isObjCInterfaceType()) {
+      Diag(ParamInfo.getSourceRange().getBegin(),
+           diag::err_object_cannot_be_passed_returned_by_value) << 0 << RetTy;
+      return;
+    }
 
     if (!RetTy->isDependentType())
-      CurBlock->ReturnType = RetTy;
+      CurBlock->ReturnType = RetTy.getTypePtr();
     return;
   }
 
@@ -4683,22 +4690,26 @@
     for (unsigned i = 0, e = FTI.NumArgs; i != e; ++i)
       CurBlock->Params.push_back(FTI.ArgInfo[i].Param.getAs<ParmVarDecl>());
     CurBlock->isVariadic = FTI.isVariadic;
-    QualType T = GetTypeForDeclarator (ParamInfo, CurScope);
-
-    Type* RetTy = T.getTypePtr()->getAsFunctionType()->getResultType()
-      .getTypePtr();
-
-    if (!RetTy->isDependentType())
-      CurBlock->ReturnType = RetTy;
   }
   CurBlock->TheDecl->setParams(Context, &CurBlock->Params[0], 
-                                        CurBlock->Params.size());
+                               CurBlock->Params.size());
 
   for (BlockDecl::param_iterator AI = CurBlock->TheDecl->param_begin(),
        E = CurBlock->TheDecl->param_end(); AI != E; ++AI)
     // If this has an identifier, add it to the scope stack.
     if ((*AI)->getIdentifier())
       PushOnScopeChains(*AI, CurBlock->TheScope);
+
+  // Analyze the return type.
+  QualType T = GetTypeForDeclarator(ParamInfo, CurScope);
+  QualType RetTy = T->getAsFunctionType()->getResultType();
+  
+  // Do not allow returning a objc interface by-value.
+  if (RetTy->isObjCInterfaceType()) {
+    Diag(ParamInfo.getSourceRange().getBegin(),
+         diag::err_object_cannot_be_passed_returned_by_value) << 0 << RetTy;
+  } else if (!RetTy->isDependentType())
+    CurBlock->ReturnType = RetTy.getTypePtr();
 }
 
 /// ActOnBlockError - If there is an error parsing a block, this callback