Added wrapper for CFReadStreamCreateForStreamedHTTPRequest().

rhaug1 · Mar 24, 2009 · 2c2da8f · 2c2da8f
1 parent c41beb8
commit 2c2da8f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/HTTPMessage/HTTPMessage.h b/HTTPMessage/HTTPMessage.h
@@ -86,6 +86,10 @@
 // returns nil if the receiver is a response message
 - (NSInputStream *) inputStream;
 
+// if you want/need to provide the message body as a stream, here's your lad.
+// if the input stream isn't required, just discard it immediately.
+- (NSInputStream *) inputStreamUsingStreamedBodyData: (NSInputStream *) bodyStream;
+
 @end
 
 // HTTP version string constants

diff --git a/HTTPMessage/HTTPMessage.m b/HTTPMessage/HTTPMessage.m
@@ -225,4 +225,13 @@ - (NSInputStream *) inputStream
 	return ( [result autorelease] );
 }
 
+- (NSInputStream *) inputStreamUsingStreamedBodyData: (NSInputStream *) bodyStream
+{
+	if ( self.isRequest == NO )
+		return ( nil );
+
+	NSInputStream * result = NSMakeCollectable( CFReadStreamCreateForStreamedHTTPRequest(kCFAllocatorDefault, _internal, (CFReadStreamRef)bodyStream) );
+	return ( [result autorelease] );
+}
+
 @end