Fixing bug in setReceiveFilter:: - forgot to release previously set ivars.

robbiehanson · robbiehanson · commit d76ac2aa7adb · 2011-12-16T16:28:26.000-08:00
diff --git a/GCD/GCDAsyncUdpSocket.m b/GCD/GCDAsyncUdpSocket.m
@@ -3957,7 +3957,7 @@ - (void)setReceiveFilter:(GCDAsyncUdpSocketReceiveFilterBlock)inFilterBlock with
 	{
 		NSAssert(inFilterQueue, @"Must provide a dispatch_queue in which to run the filter block.");
 		
-		newFilterBlock = Block_copy(inFilterBlock);
+		newFilterBlock = [inFilterBlock copy];
 		newFilterQueue = inFilterQueue;
 		dispatch_retain(newFilterQueue);
 	}
@@ -3969,6 +3969,12 @@ - (void)setReceiveFilter:(GCDAsyncUdpSocketReceiveFilterBlock)inFilterBlock with
 	
 	dispatch_block_t block = ^{
 		
+		if (filterBlock)
+			[filterBlock release];
+		
+		if (filterQueue)
+			dispatch_release(filterQueue);
+		
 		filterBlock = newFilterBlock;
 		filterQueue = newFilterQueue;
 	};
