diff options
| -rw-r--r-- | Driver/RewriteBlocks.cpp | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/Driver/RewriteBlocks.cpp b/Driver/RewriteBlocks.cpp index 83fcc03aee..83bc4280c8 100644 --- a/Driver/RewriteBlocks.cpp +++ b/Driver/RewriteBlocks.cpp @@ -1014,14 +1014,18 @@ Stmt *RewriteBlocks::RewriteFunctionBody(Stmt *S) { RewriteBlockCall(CE); } if (DeclStmt *DS = dyn_cast<DeclStmt>(S)) { - ScopedDecl *SD = DS->getDecl(); - if (ValueDecl *ND = dyn_cast<ValueDecl>(SD)) { - if (isBlockPointerType(ND->getType())) - RewriteBlockPointerDecl(ND); - } - if (TypedefDecl *TD = dyn_cast<TypedefDecl>(SD)) { - if (isBlockPointerType(TD->getUnderlyingType())) - RewriteBlockPointerDecl(TD); + for (DeclStmt::decl_iterator DI = DS->decl_begin(), DE = DS->decl_end(); + DI != DE; ++DI) { + + ScopedDecl *SD = *DI; + if (ValueDecl *ND = dyn_cast<ValueDecl>(SD)) { + if (isBlockPointerType(ND->getType())) + RewriteBlockPointerDecl(ND); + } + if (TypedefDecl *TD = dyn_cast<TypedefDecl>(SD)) { + if (isBlockPointerType(TD->getUnderlyingType())) + RewriteBlockPointerDecl(TD); + } } } // Handle specific things. |