Fix request stream exhaustion error by copying the original NSInputStream

Author: aburgel

AFNetworking/AFHTTPClient.m

@@ -740,7 +740,9 @@ - (id)copyWithZone:(NSZone *)zone {
 @interface AFHTTPBodyPart : NSObject
 @property (nonatomic, assign) NSStringEncoding stringEncoding;
 @property (nonatomic, strong) NSDictionary *headers;
-@property (nonatomic, strong) NSInputStream *inputStream;
+@property (nonatomic, strong) NSData *inputData;
+@property (nonatomic, strong) NSURL *inputURL;
+@property (nonatomic, readonly) NSInputStream *inputStream;
 @property (nonatomic, assign) unsigned long long bodyContentLength;
 
 @property (nonatomic, assign) BOOL hasInitialBoundary;
@@ -821,8 +823,8 @@ - (BOOL)appendPartWithFileURL:(NSURL *)fileURL
     AFHTTPBodyPart *bodyPart = [[AFHTTPBodyPart alloc] init];
     bodyPart.stringEncoding = self.stringEncoding;
     bodyPart.headers = mutableHeaders;
-    bodyPart.inputStream = [NSInputStream inputStreamWithURL:fileURL];
-    
+    bodyPart.inputURL = fileURL;
+
     NSDictionary *fileAttributes = [[NSFileManager defaultManager] attributesOfItemAtPath:[fileURL path] error:nil];
     bodyPart.bodyContentLength = [[fileAttributes objectForKey:NSFileSize] unsignedLongLongValue];
 
@@ -867,7 +869,7 @@ - (void)appendPartWithHeaders:(NSDictionary *)headers
     bodyPart.stringEncoding = self.stringEncoding;
     bodyPart.headers = headers;
     bodyPart.bodyContentLength = [body length];
-    bodyPart.inputStream = [NSInputStream inputStreamWithData:body];
+    bodyPart.inputData = body;
 
     [self.bodyStream appendHTTPBodyPart:bodyPart];
 }
@@ -898,7 +900,7 @@ - (NSMutableURLRequest *)requestByFinalizingMultipartFormData {
 
 #pragma mark -
 
-@interface AFMultipartBodyStream ()
+@interface AFMultipartBodyStream () <NSCopying>
 @property (nonatomic, assign) NSStreamStatus streamStatus;
 @property (nonatomic, strong) NSError *streamError;
 
@@ -931,6 +933,20 @@ - (id)initWithStringEncoding:(NSStringEncoding)encoding {
     return self;
 }
 
+-(id)copyWithZone:(NSZone *)zone {
+    AFMultipartBodyStream *bodyStreamCopy = [[[self class] allocWithZone:zone] initWithStringEncoding:self.stringEncoding];
+
+    for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts)
+    {
+        AFHTTPBodyPart *bodyPartCopy = [bodyPart copy];
+        [bodyStreamCopy appendHTTPBodyPart:bodyPartCopy];
+    }
+
+    [bodyStreamCopy setInitialAndFinalBoundaries];
+
+    return bodyStreamCopy;
+}
+
 - (void)setInitialAndFinalBoundaries {
     if ([self.HTTPBodyParts count] > 0) {
         for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts) {
@@ -1051,8 +1067,9 @@ - (BOOL)_setCFClientFlags:(__unused CFOptionFlags)inFlags
     AFFinalBoundaryPhase         = 4,
 } AFHTTPBodyPartReadPhase;
 
-@interface AFHTTPBodyPart () {
+@interface AFHTTPBodyPart () <NSCopying> {
     AFHTTPBodyPartReadPhase _phase;
+    NSInputStream *_inputStream;
     unsigned long long _phaseReadOffset;
 }
 
@@ -1063,7 +1080,8 @@ @implementation AFHTTPBodyPart
 @synthesize stringEncoding = _stringEncoding;
 @synthesize headers = _headers;
 @synthesize bodyContentLength = _bodyContentLength;
-@synthesize inputStream = _inputStream;
+@synthesize inputData = _inputData;
+@synthesize inputURL = _inputURL;
 @synthesize hasInitialBoundary = _hasInitialBoundary;
 @synthesize hasFinalBoundary = _hasFinalBoundary;
 
@@ -1078,13 +1096,39 @@ - (id)init {
     return self;
 }
 
+- (id)copyWithZone:(NSZone *)zone {
+    AFHTTPBodyPart *bodyPartCopy = [[[self class] allocWithZone:zone] init];
+
+    bodyPartCopy.stringEncoding = self.stringEncoding;
+    bodyPartCopy.headers = self.headers;
+    bodyPartCopy.bodyContentLength = self.bodyContentLength;
+    bodyPartCopy.inputData = self.inputData;
+    bodyPartCopy.inputURL = self.inputURL;
+
+    return bodyPartCopy;
+}
+
 - (void)dealloc {
     if (_inputStream) {
         [_inputStream close];
         _inputStream = nil;
     }
 }
 
+- (NSInputStream *)inputStream {
+    if (_inputStream) {
+        return _inputStream;
+    }
+
+    if (self.inputData) {
+        _inputStream = [NSInputStream inputStreamWithData:self.inputData];
+    } else if (self.inputURL) {
+        _inputStream = [NSInputStream inputStreamWithURL:self.inputURL];
+    }
+
+    return _inputStream;
+}
+
 - (NSString *)stringForHeaders {
     NSMutableString *headerString = [NSMutableString string];
     for (NSString *field in [self.headers allKeys]) {

AFNetworking/AFURLConnectionOperation.m

@@ -657,6 +657,19 @@ - (NSCachedURLResponse *)connection:(NSURLConnection *)connection
     }
 }
 
+- (NSInputStream *)connection:(NSURLConnection *)connection needNewBodyStream:(NSURLRequest *)request
+{
+    NSInputStream *newBodyStream = nil;
+    
+    NSInputStream *bodyStream = request.HTTPBodyStream;
+    if ([bodyStream conformsToProtocol:@protocol(NSCopying)])
+    {
+        newBodyStream = [bodyStream copy];
+    }
+
+    return newBodyStream;
+}
+
 #pragma mark - NSCoding
 
 - (id)initWithCoder:(NSCoder *)aDecoder {