[arcmt] Fix handling NSMakeCollectable inside an objc method.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@143980 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/ARCMigrate/Transforms.cpp b/lib/ARCMigrate/Transforms.cpp
index 86812a5..96c472c 100644
--- a/lib/ARCMigrate/Transforms.cpp
+++ b/lib/ARCMigrate/Transforms.cpp
@@ -302,6 +302,7 @@
class ASTTransform : public RecursiveASTVisitor<ASTTransform> {
MigrationContext &MigrateCtx;
+ typedef RecursiveASTVisitor<ASTTransform> base;
public:
ASTTransform(MigrationContext &MigrateCtx) : MigrateCtx(MigrateCtx) { }
@@ -315,7 +316,7 @@
E = MigrateCtx.traversers_end(); I != E; ++I)
(*I)->traverseObjCImplementation(ImplCtx);
- return true;
+ return base::TraverseObjCImplementationDecl(D);
}
bool TraverseStmt(Stmt *rootS) {
diff --git a/test/ARCMT/GC.m b/test/ARCMT/GC.m
index e49bca8..2884679 100644
--- a/test/ARCMT/GC.m
+++ b/test/ARCMT/GC.m
@@ -66,4 +66,8 @@
@implementation I4Impl
@synthesize pw1, pw2, ps, pds, pds2;
+
+-(void)test1:(CFTypeRef *)cft {
+ id x = NSMakeCollectable(cft);
+}
@end
diff --git a/test/ARCMT/GC.m.result b/test/ARCMT/GC.m.result
index 67a70e3..aaa5ee5 100644
--- a/test/ARCMT/GC.m.result
+++ b/test/ARCMT/GC.m.result
@@ -61,4 +61,8 @@
@implementation I4Impl
@synthesize pw1, pw2, ps, pds, pds2;
+
+-(void)test1:(CFTypeRef *)cft {
+ id x = CFBridgingRelease(cft);
+}
@end