X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/5c19dc3ae3bd8e40a9c028b0deddd50ff337692c..dd5fb164cf5b32c462296bc65e289e100f74b59a:/OSX/libsecurity_transform/lib/SingleShotSource.cpp?ds=inline

diff --git a/OSX/libsecurity_transform/lib/SingleShotSource.cpp b/OSX/libsecurity_transform/lib/SingleShotSource.cpp
index f5121f21..9d5f2574 100644
--- a/OSX/libsecurity_transform/lib/SingleShotSource.cpp
+++ b/OSX/libsecurity_transform/lib/SingleShotSource.cpp
@@ -1,5 +1,6 @@
 
 #include "SingleShotSource.h"
+#include "Utilities.h"
 #include <string>
 
 using namespace std;
@@ -15,7 +16,7 @@ SingleShotSource::SingleShotSource(CFTypeRef value, Transform* t, CFStringRef na
 void SingleShotSource::DoActivate()
 {
 	// Make sure our destination doesn't vanish while we are sending it data (or the final NULL)
-	CFRetain(mDestination->GetCFObject());
+	CFRetainSafe(mDestination->GetCFObject());
 	
 	// take our value and send it on its way
 	mDestination->SetAttribute(mDestinationName, GetValue());
@@ -23,7 +24,7 @@ void SingleShotSource::DoActivate()
 	// send an end of stream
 	mDestination->SetAttribute(mDestinationName, NULL);
 
-	CFRelease(mDestination->GetCFObject());
+	CFReleaseSafe(mDestination->GetCFObject());
 }