commit b30a2f32eea7f826f77149e1f7bf28cbc3b77284
author Ted Kremenek <kremenek@apple.com> Sat Apr 25 01:21:50 2009 +0000
committer Ted Kremenek <kremenek@apple.com> Sat Apr 25 01:21:50 2009 +0000
tree dbd2bde50ac02b1976711fed609a444a654303c7
parent f93cfeefb4604d8d5a6c657111659a95a5538d0b

Hook up attribute 'objc_ownership_retain' to the analyzer. This attribute allows
users to specify that a method's argument is visibly retained (reference count
incremented).


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

lib/Analysis/CFRefCount.cpp

diff --git a/lib/Analysis/CFRefCount.cpp b/lib/Analysis/CFRefCount.cpp
index 9080e66..bb957b8 100644
--- a/lib/Analysis/CFRefCount.cpp
+++ b/lib/Analysis/CFRefCount.cpp
@@ -1095,7 +1095,18 @@
     }
   }
   
-  if (!hasRetEffect)
+  // Determine if there are any arguments with a specific ArgEffect.
+  bool hasArgEffect = false;
+  unsigned i = 0;
+  for (ObjCMethodDecl::param_iterator I = MD->param_begin(),
+       E = MD->param_end(); I != E; ++I, ++i) {
+    if ((*I)->getAttr<ObjCOwnershipRetainAttr>()) {
+      ScratchArgs.push_back(std::make_pair(i, IncRefMsg));
+      hasArgEffect = true;
+    }
+}
+  
+  if (!hasRetEffect && !hasArgEffect)
     return 0;
 
   return getPersistentSummary(RE);

lib/Parse/AttributeList.cpp

diff --git a/lib/Parse/AttributeList.cpp b/lib/Parse/AttributeList.cpp
index a205c04..8048db8 100644
--- a/lib/Parse/AttributeList.cpp
+++ b/lib/Parse/AttributeList.cpp
@@ -135,7 +135,7 @@
     break;
   case 21:
     if (!memcmp(Str, "objc_ownership_retain", 21))
-      return AT_objc_ownership_returns;      
+      return AT_objc_ownership_retain;      
   case 22:
     if (!memcmp(Str, "objc_ownership_returns", 22))
       return AT_objc_ownership_returns;

test/Analysis/retain-release.m

diff --git a/test/Analysis/retain-release.m b/test/Analysis/retain-release.m
index 0de2025..ae1b75c 100644
--- a/test/Analysis/retain-release.m
+++ b/test/Analysis/retain-release.m
@@ -411,7 +411,7 @@
 
 @interface TestOwnershipAttr : NSObject
 - (NSString*) returnsAnOwnedString __attribute__((objc_ownership_returns));
-- (void) myRetain:(id __attribute__((objc_ownership_retain)))obj;
+- (void) myRetain:(id)__attribute__((objc_ownership_retain))obj;
 @end
 
 void test_attr_1(TestOwnershipAttr *X) {
@@ -419,7 +419,7 @@
 }
 
 void test_attr_2(TestOwnershipAttr *X) {
-  NSString *str = [X returnsAnOwnedString]; // no-warning (yet)
+  NSString *str = [X returnsAnOwnedString]; // expected-warning{{leak}}
   [X myRetain:str];
   [str release];
 }