[cfe-commits] r68811 - in /cfe/trunk: lib/CodeGen/CGBlocks.cpp lib/CodeGen/CGBlocks.h test/CodeGen/blocks-1.c

Mike Stump mrs at apple.com
Fri Apr 10 11:52:28 PDT 2009


Author: mrs
Date: Fri Apr 10 13:52:28 2009
New Revision: 68811

URL: http://llvm.org/viewvc/llvm-project?rev=68811&view=rev
Log:
Fixup codegen for nested blocks that use copy/dispose in the inner
blocks, so that the outer blocks use it as well.  Radar 6762279

Modified:
    cfe/trunk/lib/CodeGen/CGBlocks.cpp
    cfe/trunk/lib/CodeGen/CGBlocks.h
    cfe/trunk/test/CodeGen/blocks-1.c

Modified: cfe/trunk/lib/CodeGen/CGBlocks.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/CodeGen/CGBlocks.cpp?rev=68811&r1=68810&r2=68811&view=diff

==============================================================================
--- cfe/trunk/lib/CodeGen/CGBlocks.cpp (original)
+++ cfe/trunk/lib/CodeGen/CGBlocks.cpp Fri Apr 10 13:52:28 2009
@@ -48,10 +48,10 @@
 
   if (BlockHasCopyDispose) {
     // copy_func_helper_decl
-    Elts.push_back(BuildCopyHelper(Ty, *NoteForHelper));
+    Elts.push_back(BuildCopyHelper(Ty, NoteForHelper));
 
     // destroy_func_decl
-    Elts.push_back(BuildDestroyHelper(Ty, *NoteForHelper));
+    Elts.push_back(BuildDestroyHelper(Ty, NoteForHelper));
   }
 
   C = llvm::ConstantStruct::get(Elts);
@@ -178,7 +178,6 @@
 
     if (subBlockDeclRefDecls.size() == 0) {
       // __descriptor
-      assert(subBlockHasCopyDispose == false);
       Elts[4] = BuildDescriptorBlockDecl(subBlockHasCopyDispose, subBlockSize, 0, 0);
 
       // Optimize to being a global block.
@@ -714,7 +713,7 @@
 
 llvm::Constant *BlockFunction::
 GenerateCopyHelperFunction(bool BlockHasCopyDispose, const llvm::StructType *T,
-                           std::vector<HelperInfo> &NoteForHelper) {
+                           std::vector<HelperInfo> *NoteForHelperp) {
   QualType R = getContext().VoidTy;
 
   FunctionArgList Args;
@@ -752,31 +751,36 @@
 
   llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src);
   llvm::Type *PtrPtrT;
-  PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
-  SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT);
-  SrcObj = Builder.CreateLoad(SrcObj);
-
-  llvm::Value *DstObj = CGF.GetAddrOfLocalVar(Dst);
-  DstObj = Builder.CreateBitCast(DstObj, llvm::PointerType::get(T, 0));
-
-  for (unsigned i=0; i < NoteForHelper.size(); ++i) {
-    int flag = NoteForHelper[i].flag;
-    int index = NoteForHelper[i].index;
-
-    if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF)
-        || NoteForHelper[i].RequiresCopying) {
-      llvm::Value *Srcv = SrcObj;
-      Srcv = Builder.CreateStructGEP(Srcv, index);
-      Srcv = Builder.CreateBitCast(Srcv,
-                                   llvm::PointerType::get(PtrToInt8Ty, 0));
-      Srcv = Builder.CreateLoad(Srcv);
-
-      llvm::Value *Dstv = Builder.CreateStructGEP(DstObj, index);
-      Dstv = Builder.CreateBitCast(Dstv, PtrToInt8Ty);
-
-      llvm::Value *N = llvm::ConstantInt::get(llvm::Type::Int32Ty, flag);
-      llvm::Value *F = getBlockObjectAssign();
-      Builder.CreateCall3(F, Dstv, Srcv, N);
+
+  if (NoteForHelperp) {
+    std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp;
+
+    PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
+    SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT);
+    SrcObj = Builder.CreateLoad(SrcObj);
+
+    llvm::Value *DstObj = CGF.GetAddrOfLocalVar(Dst);
+    DstObj = Builder.CreateBitCast(DstObj, llvm::PointerType::get(T, 0));
+
+    for (unsigned i=0; i < NoteForHelper.size(); ++i) {
+      int flag = NoteForHelper[i].flag;
+      int index = NoteForHelper[i].index;
+
+      if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF)
+          || NoteForHelper[i].RequiresCopying) {
+        llvm::Value *Srcv = SrcObj;
+        Srcv = Builder.CreateStructGEP(Srcv, index);
+        Srcv = Builder.CreateBitCast(Srcv,
+                                     llvm::PointerType::get(PtrToInt8Ty, 0));
+        Srcv = Builder.CreateLoad(Srcv);
+
+        llvm::Value *Dstv = Builder.CreateStructGEP(DstObj, index);
+        Dstv = Builder.CreateBitCast(Dstv, PtrToInt8Ty);
+
+        llvm::Value *N = llvm::ConstantInt::get(llvm::Type::Int32Ty, flag);
+        llvm::Value *F = getBlockObjectAssign();
+        Builder.CreateCall3(F, Dstv, Srcv, N);
+      }
     }
   }
 
@@ -788,7 +792,7 @@
 llvm::Constant *BlockFunction::
 GenerateDestroyHelperFunction(bool BlockHasCopyDispose,
                               const llvm::StructType* T,
-                              std::vector<HelperInfo> &NoteForHelper) {
+                              std::vector<HelperInfo> *NoteForHelperp) {
   QualType R = getContext().VoidTy;
 
   FunctionArgList Args;
@@ -821,25 +825,29 @@
                                           true);
   CGF.StartFunction(FD, R, Fn, Args, SourceLocation());
 
-  llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src);
-  llvm::Type *PtrPtrT;
-  PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
-  SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT);
-  SrcObj = Builder.CreateLoad(SrcObj);
-
-  for (unsigned i=0; i < NoteForHelper.size(); ++i) {
-    int flag = NoteForHelper[i].flag;
-    int index = NoteForHelper[i].index;
-
-    if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF)
-        || NoteForHelper[i].RequiresCopying) {
-      llvm::Value *Srcv = SrcObj;
-      Srcv = Builder.CreateStructGEP(Srcv, index);
-      Srcv = Builder.CreateBitCast(Srcv,
-                                   llvm::PointerType::get(PtrToInt8Ty, 0));
-      Srcv = Builder.CreateLoad(Srcv);
+  if (NoteForHelperp) {
+    std::vector<HelperInfo> &NoteForHelper = *NoteForHelperp;
+
+    llvm::Value *SrcObj = CGF.GetAddrOfLocalVar(Src);
+    llvm::Type *PtrPtrT;
+    PtrPtrT = llvm::PointerType::get(llvm::PointerType::get(T, 0), 0);
+    SrcObj = Builder.CreateBitCast(SrcObj, PtrPtrT);
+    SrcObj = Builder.CreateLoad(SrcObj);
+
+    for (unsigned i=0; i < NoteForHelper.size(); ++i) {
+      int flag = NoteForHelper[i].flag;
+      int index = NoteForHelper[i].index;
+
+      if ((NoteForHelper[i].flag & BLOCK_FIELD_IS_BYREF)
+          || NoteForHelper[i].RequiresCopying) {
+        llvm::Value *Srcv = SrcObj;
+        Srcv = Builder.CreateStructGEP(Srcv, index);
+        Srcv = Builder.CreateBitCast(Srcv,
+                                     llvm::PointerType::get(PtrToInt8Ty, 0));
+        Srcv = Builder.CreateLoad(Srcv);
 
-      BuildBlockRelease(Srcv, flag);
+        BuildBlockRelease(Srcv, flag);
+      }
     }
   }
 
@@ -849,15 +857,15 @@
 }
 
 llvm::Constant *BlockFunction::BuildCopyHelper(const llvm::StructType *T,
-                                       std::vector<HelperInfo> &NoteForHelper) {
+                                       std::vector<HelperInfo> *NoteForHelper) {
   return CodeGenFunction(CGM).GenerateCopyHelperFunction(BlockHasCopyDispose,
                                                          T, NoteForHelper);
 }
 
 llvm::Constant *BlockFunction::BuildDestroyHelper(const llvm::StructType *T,
-                                       std::vector<HelperInfo> &NoteForHelper) {
+                                      std::vector<HelperInfo> *NoteForHelperp) {
   return CodeGenFunction(CGM).GenerateDestroyHelperFunction(BlockHasCopyDispose,
-                                                            T, NoteForHelper);
+                                                            T, NoteForHelperp);
 }
 
 llvm::Constant *BlockFunction::

Modified: cfe/trunk/lib/CodeGen/CGBlocks.h
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/CodeGen/CGBlocks.h?rev=68811&r1=68810&r2=68811&view=diff

==============================================================================
--- cfe/trunk/lib/CodeGen/CGBlocks.h (original)
+++ cfe/trunk/lib/CodeGen/CGBlocks.h Fri Apr 10 13:52:28 2009
@@ -183,14 +183,14 @@
   ImplicitParamDecl *getBlockStructDecl() { return BlockStructDecl; }
 
   llvm::Constant *GenerateCopyHelperFunction(bool, const llvm::StructType *,
-                                             std::vector<HelperInfo> &);
+                                             std::vector<HelperInfo> *);
   llvm::Constant *GenerateDestroyHelperFunction(bool, const llvm::StructType *,
-                                                std::vector<HelperInfo> &);
+                                                std::vector<HelperInfo> *);
 
   llvm::Constant *BuildCopyHelper(const llvm::StructType *,
-                                  std::vector<HelperInfo> &);
+                                  std::vector<HelperInfo> *);
   llvm::Constant *BuildDestroyHelper(const llvm::StructType *,
-                                     std::vector<HelperInfo> &);
+                                     std::vector<HelperInfo> *);
 
   llvm::Constant *GeneratebyrefCopyHelperFunction(const llvm::Type *, int flag);
   llvm::Constant *GeneratebyrefDestroyHelperFunction(const llvm::Type *T, int);

Modified: cfe/trunk/test/CodeGen/blocks-1.c
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/test/CodeGen/blocks-1.c?rev=68811&r1=68810&r2=68811&view=diff

==============================================================================
--- cfe/trunk/test/CodeGen/blocks-1.c (original)
+++ cfe/trunk/test/CodeGen/blocks-1.c Fri Apr 10 13:52:28 2009
@@ -1,11 +1,11 @@
 // RUN: clang-cc %s -emit-llvm -o %t -fblocks &&
-// RUN: grep "_Block_object_dispose" %t | count 15 &&
-// RUN: grep "__copy_helper_block_" %t | count 12 &&
-// RUN: grep "__destroy_helper_block_" %t | count 12 &&
+// RUN: grep "_Block_object_dispose" %t | count 17 &&
+// RUN: grep "__copy_helper_block_" %t | count 16 &&
+// RUN: grep "__destroy_helper_block_" %t | count 16 &&
 // RUN: grep "__Block_byref_id_object_copy_" %t | count 2 &&
 // RUN: grep "__Block_byref_id_object_dispose_" %t | count 2 &&
 // RUN: grep "i32 135)" %t | count 2 &&
-// RUN: grep "_Block_object_assign" %t | count 9
+// RUN: grep "_Block_object_assign" %t | count 10
 
 #include <stdio.h>
 
@@ -57,7 +57,14 @@
 void test6() {
   __block int i;
   ^{ i=1; }();
- ^{};
+  ^{}();
+}
+
+void test7() {
+  ^{
+    __block int i;
+    ^{ i = 1; }();
+  }();
 }
 
 int main() {





More information about the cfe-commits mailing list