// AFHTTPClient.m // // Copyright (c) 2011 Gowalla (http://gowalla.com/) // // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to deal // in the Software without restriction, including without limitation the rights // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell // copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // // The above copyright notice and this permission notice shall be included in // all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN // THE SOFTWARE. #import #import "AFHTTPClient.h" #import "AFHTTPRequestOperation.h" #import "AFJSONUtilities.h" #import #if __IPHONE_OS_VERSION_MIN_REQUIRED #import #import #else #import #endif #ifdef _SYSTEMCONFIGURATION_H #import #import #import #import #import #import #endif NSString * const AFNetworkingReachabilityDidChangeNotification = @"com.alamofire.networking.reachability.change"; @interface AFStreamingMultipartFormData : NSObject - (id)initWithURLRequest:(NSMutableURLRequest *)request stringEncoding:(NSStringEncoding)encoding; - (NSMutableURLRequest *)requestByFinalizingMultipartFormData; @end @interface AFMultipartBodyStream : NSInputStream - (id)initWithStringEncoding:(NSStringEncoding)encoding; - (BOOL)addFileFromURL:(NSURL *)fileURL name:(NSString *)name error:(NSError **)error; - (void)addFormData:(NSData *)data name:(NSString *)name; - (BOOL)empty; - (NSUInteger)contentLength; @end #pragma mark - #ifdef _SYSTEMCONFIGURATION_H typedef SCNetworkReachabilityRef AFNetworkReachabilityRef; typedef void (^AFNetworkReachabilityStatusBlock)(AFNetworkReachabilityStatus status); #else typedef id AFNetworkReachabilityRef; #endif typedef void (^AFCompletionBlock)(void); static NSString * AFBase64EncodedStringFromString(NSString *string) { NSData *data = [NSData dataWithBytes:[string UTF8String] length:[string lengthOfBytesUsingEncoding:NSUTF8StringEncoding]]; NSUInteger length = [data length]; NSMutableData *mutableData = [NSMutableData dataWithLength:((length + 2) / 3) * 4]; uint8_t *input = (uint8_t *)[data bytes]; uint8_t *output = (uint8_t *)[mutableData mutableBytes]; for (NSUInteger i = 0; i < length; i += 3) { NSUInteger value = 0; for (NSUInteger j = i; j < (i + 3); j++) { value <<= 8; if (j < length) { value |= (0xFF & input[j]); } } static uint8_t const kAFBase64EncodingTable[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"; NSUInteger idx = (i / 3) * 4; output[idx + 0] = kAFBase64EncodingTable[(value >> 18) & 0x3F]; output[idx + 1] = kAFBase64EncodingTable[(value >> 12) & 0x3F]; output[idx + 2] = (i + 1) < length ? kAFBase64EncodingTable[(value >> 6) & 0x3F] : '='; output[idx + 3] = (i + 2) < length ? kAFBase64EncodingTable[(value >> 0) & 0x3F] : '='; } return [[[NSString alloc] initWithData:mutableData encoding:NSASCIIStringEncoding] autorelease]; } NSString * AFURLEncodedStringFromStringWithEncoding(NSString *string, NSStringEncoding encoding) { static NSString * const kAFLegalCharactersToBeEscaped = @"?!@#$^&%*+=,:;'\"`<>()[]{}/\\|~ "; return [(NSString *)CFURLCreateStringByAddingPercentEscapes(kCFAllocatorDefault, (CFStringRef)string, NULL, (CFStringRef)kAFLegalCharactersToBeEscaped, CFStringConvertNSStringEncodingToEncoding(encoding)) autorelease]; } #pragma mark - @interface AFQueryStringComponent : NSObject { @private NSString *_key; NSString *_value; } @property (readwrite, nonatomic, retain) id key; @property (readwrite, nonatomic, retain) id value; - (id)initWithKey:(id)key value:(id)value; - (NSString *)URLEncodedStringValueWithEncoding:(NSStringEncoding)stringEncoding; @end @implementation AFQueryStringComponent @synthesize key = _key; @synthesize value = _value; - (id)initWithKey:(id)key value:(id)value { self = [super init]; if (!self) { return nil; } self.key = key; self.value = value; return self; } - (void)dealloc { [_key release]; [_value release]; [super dealloc]; } - (NSString *)URLEncodedStringValueWithEncoding:(NSStringEncoding)stringEncoding { return [NSString stringWithFormat:@"%@=%@", self.key, AFURLEncodedStringFromStringWithEncoding([self.value description], stringEncoding)]; } @end #pragma mark - extern NSArray * AFQueryStringComponentsFromKeyAndValue(NSString *key, id value); extern NSArray * AFQueryStringComponentsFromKeyAndDictionaryValue(NSString *key, NSDictionary *value); extern NSArray * AFQueryStringComponentsFromKeyAndArrayValue(NSString *key, NSArray *value); NSString * AFQueryStringFromParametersWithEncoding(NSDictionary *parameters, NSStringEncoding stringEncoding) { NSMutableArray *mutableComponents = [NSMutableArray array]; for (AFQueryStringComponent *component in AFQueryStringComponentsFromKeyAndValue(nil, parameters)) { [mutableComponents addObject:[component URLEncodedStringValueWithEncoding:stringEncoding]]; } return [mutableComponents componentsJoinedByString:@"&"]; } NSArray * AFQueryStringComponentsFromKeyAndValue(NSString *key, id value) { NSMutableArray *mutableQueryStringComponents = [NSMutableArray array]; if([value isKindOfClass:[NSDictionary class]]) { [mutableQueryStringComponents addObjectsFromArray:AFQueryStringComponentsFromKeyAndDictionaryValue(key, value)]; } else if([value isKindOfClass:[NSArray class]]) { [mutableQueryStringComponents addObjectsFromArray:AFQueryStringComponentsFromKeyAndArrayValue(key, value)]; } else { [mutableQueryStringComponents addObject:[[[AFQueryStringComponent alloc] initWithKey:key value:value] autorelease]]; } return mutableQueryStringComponents; } NSArray * AFQueryStringComponentsFromKeyAndDictionaryValue(NSString *key, NSDictionary *value){ NSMutableArray *mutableQueryStringComponents = [NSMutableArray array]; [value enumerateKeysAndObjectsUsingBlock:^(id nestedKey, id nestedValue, BOOL *stop) { [mutableQueryStringComponents addObjectsFromArray:AFQueryStringComponentsFromKeyAndValue((key ? [NSString stringWithFormat:@"%@[%@]", key, nestedKey] : nestedKey), nestedValue)]; }]; return mutableQueryStringComponents; } NSArray * AFQueryStringComponentsFromKeyAndArrayValue(NSString *key, NSArray *value) { NSMutableArray *mutableQueryStringComponents = [NSMutableArray array]; [value enumerateObjectsUsingBlock:^(id nestedValue, NSUInteger idx, BOOL *stop) { [mutableQueryStringComponents addObjectsFromArray:AFQueryStringComponentsFromKeyAndValue([NSString stringWithFormat:@"%@[]", key], nestedValue)]; }]; return mutableQueryStringComponents; } static NSString * AFJSONStringFromParameters(NSDictionary *parameters) { NSError *error = nil; NSData *JSONData = AFJSONEncode(parameters, &error); if (!error) { return [[[NSString alloc] initWithData:JSONData encoding:NSUTF8StringEncoding] autorelease]; } else { return nil; } } static NSString * AFPropertyListStringFromParameters(NSDictionary *parameters) { NSString *propertyListString = nil; NSError *error = nil; NSData *propertyListData = [NSPropertyListSerialization dataWithPropertyList:parameters format:NSPropertyListXMLFormat_v1_0 options:0 error:&error]; if (!error) { propertyListString = [[[NSString alloc] initWithData:propertyListData encoding:NSUTF8StringEncoding] autorelease]; } return propertyListString; } @interface AFHTTPClient () @property (readwrite, nonatomic, retain) NSURL *baseURL; @property (readwrite, nonatomic, retain) NSMutableArray *registeredHTTPOperationClassNames; @property (readwrite, nonatomic, retain) NSMutableDictionary *defaultHeaders; @property (readwrite, nonatomic, retain) NSOperationQueue *operationQueue; #ifdef _SYSTEMCONFIGURATION_H @property (readwrite, nonatomic, assign) AFNetworkReachabilityRef networkReachability; @property (readwrite, nonatomic, assign) AFNetworkReachabilityStatus networkReachabilityStatus; @property (readwrite, nonatomic, copy) AFNetworkReachabilityStatusBlock networkReachabilityStatusBlock; #endif #ifdef _SYSTEMCONFIGURATION_H - (void)startMonitoringNetworkReachability; - (void)stopMonitoringNetworkReachability; #endif @end @implementation AFHTTPClient @synthesize baseURL = _baseURL; @synthesize stringEncoding = _stringEncoding; @synthesize parameterEncoding = _parameterEncoding; @synthesize registeredHTTPOperationClassNames = _registeredHTTPOperationClassNames; @synthesize defaultHeaders = _defaultHeaders; @synthesize operationQueue = _operationQueue; #ifdef _SYSTEMCONFIGURATION_H @synthesize networkReachability = _networkReachability; @synthesize networkReachabilityStatus = _networkReachabilityStatus; @synthesize networkReachabilityStatusBlock = _networkReachabilityStatusBlock; #endif + (AFHTTPClient *)clientWithBaseURL:(NSURL *)url { return [[[self alloc] initWithBaseURL:url] autorelease]; } - (id)initWithBaseURL:(NSURL *)url { self = [super init]; if (!self) { return nil; } // Ensure terminal slash for baseURL path, so that NSURL +URLWithString:relativeToURL: works as expected if ([[url path] length] > 0 && ![[url absoluteString] hasSuffix:@"/"]) { url = [url URLByAppendingPathComponent:@""]; } self.baseURL = url; self.stringEncoding = NSUTF8StringEncoding; self.parameterEncoding = AFFormURLParameterEncoding; self.registeredHTTPOperationClassNames = [NSMutableArray array]; self.defaultHeaders = [NSMutableDictionary dictionary]; // Accept-Encoding HTTP Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.3 [self setDefaultHeader:@"Accept-Encoding" value:@"gzip"]; // Accept-Language HTTP Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.4 NSString *preferredLanguageCodes = [[NSLocale preferredLanguages] componentsJoinedByString:@", "]; [self setDefaultHeader:@"Accept-Language" value:[NSString stringWithFormat:@"%@, en-us;q=0.8", preferredLanguageCodes]]; #if __IPHONE_OS_VERSION_MIN_REQUIRED // User-Agent Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.43 [self setDefaultHeader:@"User-Agent" value:[NSString stringWithFormat:@"%@/%@ (%@, %@ %@, %@, Scale/%f)", [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleIdentifierKey], [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleVersionKey], @"unknown", [[UIDevice currentDevice] systemName], [[UIDevice currentDevice] systemVersion], [[UIDevice currentDevice] model], ([[UIScreen mainScreen] respondsToSelector:@selector(scale)] ? [[UIScreen mainScreen] scale] : 1.0)]]; #elif __MAC_OS_X_VERSION_MIN_REQUIRED [self setDefaultHeader:@"User-Agent" value:[NSString stringWithFormat:@"%@/%@ (%@)", [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleIdentifierKey], [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleVersionKey], @"unknown"]]; #endif #ifdef _SYSTEMCONFIGURATION_H self.networkReachabilityStatus = AFNetworkReachabilityStatusUnknown; [self startMonitoringNetworkReachability]; #endif self.operationQueue = [[[NSOperationQueue alloc] init] autorelease]; [self.operationQueue setMaxConcurrentOperationCount:NSOperationQueueDefaultMaxConcurrentOperationCount]; return self; } - (void)dealloc { #ifdef _SYSTEMCONFIGURATION_H [self stopMonitoringNetworkReachability]; [_networkReachabilityStatusBlock release]; #endif [_baseURL release]; [_registeredHTTPOperationClassNames release]; [_defaultHeaders release]; [_operationQueue release]; [super dealloc]; } - (NSString *)description { return [NSString stringWithFormat:@"<%@: %p, baseURL: %@, defaultHeaders: %@, registeredOperationClasses: %@, operationQueue: %@>", NSStringFromClass([self class]), self, [self.baseURL absoluteString], self.defaultHeaders, self.registeredHTTPOperationClassNames, self.operationQueue]; } #pragma mark - #ifdef _SYSTEMCONFIGURATION_H static BOOL AFURLHostIsIPAddress(NSURL *url) { struct sockaddr_in sa_in; struct sockaddr_in6 sa_in6; return [url host] && (inet_pton(AF_INET, [[url host] UTF8String], &sa_in) == 1 || inet_pton(AF_INET6, [[url host] UTF8String], &sa_in6) == 1); } static AFNetworkReachabilityStatus AFNetworkReachabilityStatusForFlags(SCNetworkReachabilityFlags flags) { BOOL isReachable = ((flags & kSCNetworkReachabilityFlagsReachable) != 0); BOOL needsConnection = ((flags & kSCNetworkReachabilityFlagsConnectionRequired) != 0); BOOL isNetworkReachable = (isReachable && !needsConnection); AFNetworkReachabilityStatus status = AFNetworkReachabilityStatusUnknown; if(isNetworkReachable == NO){ status = AFNetworkReachabilityStatusNotReachable; } #if TARGET_OS_IPHONE else if((flags & kSCNetworkReachabilityFlagsIsWWAN) != 0){ status = AFNetworkReachabilityStatusReachableViaWWAN; } #endif else { status = AFNetworkReachabilityStatusReachableViaWiFi; } return status; } static void AFNetworkReachabilityCallback(SCNetworkReachabilityRef __unused target, SCNetworkReachabilityFlags flags, void *info) { AFNetworkReachabilityStatus status = AFNetworkReachabilityStatusForFlags(flags); AFNetworkReachabilityStatusBlock block = (AFNetworkReachabilityStatusBlock)info; if (block) { block(status); } [[NSNotificationCenter defaultCenter] postNotificationName:AFNetworkingReachabilityDidChangeNotification object:[NSNumber numberWithInt:status]]; } static const void * AFNetworkReachabilityRetainCallback(const void *info) { return [(AFNetworkReachabilityStatusBlock)info copy]; } static void AFNetworkReachabilityReleaseCallback(const void *info) { [(AFNetworkReachabilityStatusBlock)info release]; } - (void)startMonitoringNetworkReachability { [self stopMonitoringNetworkReachability]; if (!self.baseURL) { return; } self.networkReachability = SCNetworkReachabilityCreateWithName(kCFAllocatorDefault, [[self.baseURL host] UTF8String]); AFNetworkReachabilityStatusBlock callback = ^(AFNetworkReachabilityStatus status){ self.networkReachabilityStatus = status; if (self.networkReachabilityStatusBlock) { self.networkReachabilityStatusBlock(status); } }; SCNetworkReachabilityContext context = {0, callback, AFNetworkReachabilityRetainCallback, AFNetworkReachabilityReleaseCallback, NULL}; SCNetworkReachabilitySetCallback(self.networkReachability, AFNetworkReachabilityCallback, &context); SCNetworkReachabilityScheduleWithRunLoop(self.networkReachability, CFRunLoopGetMain(), (CFStringRef)NSRunLoopCommonModes); /* Network reachability monitoring does not establish a baseline for IP addresses as it does for hostnames, so if the base URL host is an IP address, the initial reachability callback is manually triggered. */ if (AFURLHostIsIPAddress(self.baseURL)) { SCNetworkReachabilityFlags flags; SCNetworkReachabilityGetFlags(self.networkReachability, &flags); dispatch_async(dispatch_get_main_queue(), ^{ AFNetworkReachabilityStatus status = AFNetworkReachabilityStatusForFlags(flags); callback(status); }); } } - (void)stopMonitoringNetworkReachability { if (_networkReachability) { SCNetworkReachabilityUnscheduleFromRunLoop(_networkReachability, CFRunLoopGetMain(), (CFStringRef)NSRunLoopCommonModes); CFRelease(_networkReachability); } } - (void)setReachabilityStatusChangeBlock:(void (^)(AFNetworkReachabilityStatus status))block { self.networkReachabilityStatusBlock = block; } #endif #pragma mark - - (BOOL)registerHTTPOperationClass:(Class)operationClass { if (![operationClass isSubclassOfClass:[AFHTTPRequestOperation class]]) { return NO; } NSString *className = NSStringFromClass(operationClass); [self.registeredHTTPOperationClassNames removeObject:className]; [self.registeredHTTPOperationClassNames insertObject:className atIndex:0]; return YES; } - (void)unregisterHTTPOperationClass:(Class)operationClass { NSString *className = NSStringFromClass(operationClass); [self.registeredHTTPOperationClassNames removeObject:className]; } #pragma mark - - (NSString *)defaultValueForHeader:(NSString *)header { return [self.defaultHeaders valueForKey:header]; } - (void)setDefaultHeader:(NSString *)header value:(NSString *)value { [self.defaultHeaders setValue:value forKey:header]; } - (void)setAuthorizationHeaderWithUsername:(NSString *)username password:(NSString *)password { NSString *basicAuthCredentials = [NSString stringWithFormat:@"%@:%@", username, password]; [self setDefaultHeader:@"Authorization" value:[NSString stringWithFormat:@"Basic %@", AFBase64EncodedStringFromString(basicAuthCredentials)]]; } - (void)setAuthorizationHeaderWithToken:(NSString *)token { [self setDefaultHeader:@"Authorization" value:[NSString stringWithFormat:@"Token token=\"%@\"", token]]; } - (void)clearAuthorizationHeader { [self.defaultHeaders removeObjectForKey:@"Authorization"]; } #pragma mark - - (NSMutableURLRequest *)requestWithMethod:(NSString *)method path:(NSString *)path parameters:(NSDictionary *)parameters { NSURL *url = [NSURL URLWithString:path relativeToURL:self.baseURL]; NSMutableURLRequest *request = [[[NSMutableURLRequest alloc] initWithURL:url] autorelease]; [request setHTTPMethod:method]; [request setAllHTTPHeaderFields:self.defaultHeaders]; if (parameters) { if ([method isEqualToString:@"GET"] || [method isEqualToString:@"HEAD"] || [method isEqualToString:@"DELETE"]) { url = [NSURL URLWithString:[[url absoluteString] stringByAppendingFormat:[path rangeOfString:@"?"].location == NSNotFound ? @"?%@" : @"&%@", AFQueryStringFromParametersWithEncoding(parameters, self.stringEncoding)]]; [request setURL:url]; } else { NSString *charset = (NSString *)CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(self.stringEncoding)); switch (self.parameterEncoding) { case AFFormURLParameterEncoding:; [request setValue:[NSString stringWithFormat:@"application/x-www-form-urlencoded; charset=%@", charset] forHTTPHeaderField:@"Content-Type"]; [request setHTTPBody:[AFQueryStringFromParametersWithEncoding(parameters, self.stringEncoding) dataUsingEncoding:self.stringEncoding]]; break; case AFJSONParameterEncoding:; [request setValue:[NSString stringWithFormat:@"application/json; charset=%@", charset] forHTTPHeaderField:@"Content-Type"]; [request setHTTPBody:[AFJSONStringFromParameters(parameters) dataUsingEncoding:self.stringEncoding]]; break; case AFPropertyListParameterEncoding:; [request setValue:[NSString stringWithFormat:@"application/x-plist; charset=%@", charset] forHTTPHeaderField:@"Content-Type"]; [request setHTTPBody:[AFPropertyListStringFromParameters(parameters) dataUsingEncoding:self.stringEncoding]]; break; } } } return request; } - (NSMutableURLRequest *)multipartFormRequestWithMethod:(NSString *)method path:(NSString *)path parameters:(NSDictionary *)parameters constructingBodyWithBlock:(void (^)(id formData))block { NSMutableURLRequest *request = [self requestWithMethod:method path:path parameters:nil]; __block AFStreamingMultipartFormData * formData = [[[AFStreamingMultipartFormData alloc] initWithURLRequest:request stringEncoding:self.stringEncoding] autorelease]; // __block AFMultipartFormData *formData = [[[AFMultipartFormData alloc] initWithURLRequest:request stringEncoding:self.stringEncoding] autorelease]; for (AFQueryStringComponent *component in AFQueryStringComponentsFromKeyAndValue(nil, parameters)) { NSData *data = nil; if ([component.value isKindOfClass:[NSData class]]) { data = component.value; } else { data = [[component.value description] dataUsingEncoding:self.stringEncoding]; } if (data) { [formData appendPartWithFormData:data name:[component.key description]]; } } if (block) { block(formData); } return [formData requestByFinalizingMultipartFormData]; } - (AFHTTPRequestOperation *)HTTPRequestOperationWithRequest:(NSURLRequest *)urlRequest success:(void (^)(AFHTTPRequestOperation *operation, id responseObject))success failure:(void (^)(AFHTTPRequestOperation *operation, NSError *error))failure { AFHTTPRequestOperation *operation = nil; NSString *className = nil; NSEnumerator *enumerator = [self.registeredHTTPOperationClassNames reverseObjectEnumerator]; while (!operation && (className = [enumerator nextObject])) { Class op_class = NSClassFromString(className); if (op_class && [op_class canProcessRequest:urlRequest]) { operation = [[(AFHTTPRequestOperation *)[op_class alloc] initWithRequest:urlRequest] autorelease]; } } if (!operation) { operation = [[[AFHTTPRequestOperation alloc] initWithRequest:urlRequest] autorelease]; } [operation setCompletionBlockWithSuccess:success failure:failure]; return operation; } #pragma mark - - (void)enqueueHTTPRequestOperation:(AFHTTPRequestOperation *)operation { [self.operationQueue addOperation:operation]; } - (void)cancelAllHTTPOperationsWithMethod:(NSString *)method path:(NSString *)path { for (NSOperation *operation in [self.operationQueue operations]) { if (![operation isKindOfClass:[AFHTTPRequestOperation class]]) { continue; } if ((!method || [method isEqualToString:[[(AFHTTPRequestOperation *)operation request] HTTPMethod]]) && [path isEqualToString:[[[(AFHTTPRequestOperation *)operation request] URL] path]]) { [operation cancel]; } } } - (void)enqueueBatchOfHTTPRequestOperationsWithRequests:(NSArray *)requests progressBlock:(void (^)(NSUInteger numberOfCompletedOperations, NSUInteger totalNumberOfOperations))progressBlock completionBlock:(void (^)(NSArray *operations))completionBlock { NSMutableArray *mutableOperations = [NSMutableArray array]; for (NSURLRequest *request in requests) { AFHTTPRequestOperation *operation = [self HTTPRequestOperationWithRequest:request success:nil failure:nil]; [mutableOperations addObject:operation]; } [self enqueueBatchOfHTTPRequestOperations:mutableOperations progressBlock:progressBlock completionBlock:completionBlock]; } - (void)enqueueBatchOfHTTPRequestOperations:(NSArray *)operations progressBlock:(void (^)(NSUInteger numberOfCompletedOperations, NSUInteger totalNumberOfOperations))progressBlock completionBlock:(void (^)(NSArray *operations))completionBlock { __block dispatch_group_t dispatchGroup = dispatch_group_create(); NSBlockOperation *batchedOperation = [NSBlockOperation blockOperationWithBlock:^{ dispatch_group_notify(dispatchGroup, dispatch_get_main_queue(), ^{ if (completionBlock) { completionBlock(operations); } }); dispatch_release(dispatchGroup); }]; NSPredicate *finishedOperationPredicate = [NSPredicate predicateWithFormat:@"isFinished == YES"]; for (AFHTTPRequestOperation *operation in operations) { AFCompletionBlock originalCompletionBlock = [[operation.completionBlock copy] autorelease]; operation.completionBlock = ^{ dispatch_queue_t queue = operation.successCallbackQueue ? operation.successCallbackQueue : dispatch_get_main_queue(); dispatch_group_async(dispatchGroup, queue, ^{ if (originalCompletionBlock) { originalCompletionBlock(); } if (progressBlock) { progressBlock([[operations filteredArrayUsingPredicate:finishedOperationPredicate] count], [operations count]); } dispatch_group_leave(dispatchGroup); }); }; dispatch_group_enter(dispatchGroup); [batchedOperation addDependency:operation]; [self enqueueHTTPRequestOperation:operation]; } [self.operationQueue addOperation:batchedOperation]; } #pragma mark - - (void)getPath:(NSString *)path parameters:(NSDictionary *)parameters success:(void (^)(AFHTTPRequestOperation *operation, id responseObject))success failure:(void (^)(AFHTTPRequestOperation *operation, NSError *error))failure { NSURLRequest *request = [self requestWithMethod:@"GET" path:path parameters:parameters]; AFHTTPRequestOperation *operation = [self HTTPRequestOperationWithRequest:request success:success failure:failure]; [self enqueueHTTPRequestOperation:operation]; } - (void)postPath:(NSString *)path parameters:(NSDictionary *)parameters success:(void (^)(AFHTTPRequestOperation *operation, id responseObject))success failure:(void (^)(AFHTTPRequestOperation *operation, NSError *error))failure { NSURLRequest *request = [self requestWithMethod:@"POST" path:path parameters:parameters]; AFHTTPRequestOperation *operation = [self HTTPRequestOperationWithRequest:request success:success failure:failure]; [self enqueueHTTPRequestOperation:operation]; } - (void)putPath:(NSString *)path parameters:(NSDictionary *)parameters success:(void (^)(AFHTTPRequestOperation *operation, id responseObject))success failure:(void (^)(AFHTTPRequestOperation *operation, NSError *error))failure { NSURLRequest *request = [self requestWithMethod:@"PUT" path:path parameters:parameters]; AFHTTPRequestOperation *operation = [self HTTPRequestOperationWithRequest:request success:success failure:failure]; [self enqueueHTTPRequestOperation:operation]; } - (void)deletePath:(NSString *)path parameters:(NSDictionary *)parameters success:(void (^)(AFHTTPRequestOperation *operation, id responseObject))success failure:(void (^)(AFHTTPRequestOperation *operation, NSError *error))failure { NSURLRequest *request = [self requestWithMethod:@"DELETE" path:path parameters:parameters]; AFHTTPRequestOperation *operation = [self HTTPRequestOperationWithRequest:request success:success failure:failure]; [self enqueueHTTPRequestOperation:operation]; } - (void)patchPath:(NSString *)path parameters:(NSDictionary *)parameters success:(void (^)(AFHTTPRequestOperation *operation, id responseObject))success failure:(void (^)(AFHTTPRequestOperation *operation, NSError *error))failure { NSURLRequest *request = [self requestWithMethod:@"PATCH" path:path parameters:parameters]; AFHTTPRequestOperation *operation = [self HTTPRequestOperationWithRequest:request success:success failure:failure]; [self enqueueHTTPRequestOperation:operation]; } #pragma mark - NSCoding - (id)initWithCoder:(NSCoder *)aDecoder { NSURL *baseURL = [aDecoder decodeObjectForKey:@"baseURL"]; self = [self initWithBaseURL:baseURL]; if (!self) { return nil; } self.stringEncoding = [aDecoder decodeIntegerForKey:@"stringEncoding"]; self.parameterEncoding = [aDecoder decodeIntegerForKey:@"parameterEncoding"]; self.registeredHTTPOperationClassNames = [aDecoder decodeObjectForKey:@"registeredHTTPOperationClassNames"]; self.defaultHeaders = [aDecoder decodeObjectForKey:@"defaultHeaders"]; return self; } - (void)encodeWithCoder:(NSCoder *)aCoder { [aCoder encodeObject:self.baseURL forKey:@"baseURL"]; [aCoder encodeInteger:self.stringEncoding forKey:@"stringEncoding"]; [aCoder encodeInteger:self.parameterEncoding forKey:@"parameterEncoding"]; [aCoder encodeObject:self.registeredHTTPOperationClassNames forKey:@"registeredHTTPOperationClassNames"]; [aCoder encodeObject:self.defaultHeaders forKey:@"defaultHeaders"]; } #pragma mark - NSCopying - (id)copyWithZone:(NSZone *)zone { AFHTTPClient *HTTPClient = [[[self class] allocWithZone:zone] initWithBaseURL:self.baseURL]; HTTPClient.stringEncoding = self.stringEncoding; HTTPClient.parameterEncoding = self.parameterEncoding; HTTPClient.registeredHTTPOperationClassNames = [[self.registeredHTTPOperationClassNames copyWithZone:zone] autorelease]; HTTPClient.defaultHeaders = [[self.defaultHeaders copyWithZone:zone] autorelease]; #ifdef _SYSTEMCONFIGURATION_H HTTPClient.networkReachabilityStatusBlock = self.networkReachabilityStatusBlock; #endif return HTTPClient; } @end #pragma mark - static NSString * const kAFMultipartTemporaryFileDirectoryName = @"com.alamofire.uploads"; static NSString * AFMultipartTemporaryFileDirectoryPath() { static NSString *multipartTemporaryFilePath = nil; static dispatch_once_t onceToken; dispatch_once(&onceToken, ^{ multipartTemporaryFilePath = [[NSTemporaryDirectory() stringByAppendingPathComponent:kAFMultipartTemporaryFileDirectoryName] copy]; NSError *error = nil; if(![[NSFileManager defaultManager] createDirectoryAtPath:multipartTemporaryFilePath withIntermediateDirectories:YES attributes:nil error:&error]) { NSLog(@"Failed to create multipary temporary file directory at %@", multipartTemporaryFilePath); } }); return multipartTemporaryFilePath; } static NSString * const kAFMultipartFormBoundary = @"Boundary+0xAbCdEfGbOuNdArY"; static NSString * const kAFMultipartFormCRLF = @"\r\n"; static NSInteger const kAFStreamToStreamBufferSize = 1024*1024; //1 meg default static inline NSString * AFMultipartFormInitialBoundary() { return [NSString stringWithFormat:@"--%@%@", kAFMultipartFormBoundary, kAFMultipartFormCRLF]; } static inline NSString * AFMultipartFormEncapsulationBoundary() { return [NSString stringWithFormat:@"%@--%@%@", kAFMultipartFormCRLF, kAFMultipartFormBoundary, kAFMultipartFormCRLF]; } static inline NSString * AFMultipartFormFinalBoundary() { return [NSString stringWithFormat:@"%@--%@--%@", kAFMultipartFormCRLF, kAFMultipartFormBoundary, kAFMultipartFormCRLF]; } #pragma mark -- #pragma mark AFStreamingMultipartFormData @interface AFStreamingMultipartFormData () { AFMultipartBodyStream * bodyStream; } @property (readwrite, nonatomic, retain) NSMutableURLRequest *request; @property (readwrite, nonatomic, assign) NSStringEncoding stringEncoding; @end @implementation AFStreamingMultipartFormData @synthesize request = _request; @synthesize stringEncoding = _stringEncoding; - (id)initWithURLRequest:(NSMutableURLRequest *)request stringEncoding:(NSStringEncoding)encoding { self = [super init]; if (!self) { return nil; } self.request = request; self.stringEncoding = encoding; bodyStream = [[AFMultipartBodyStream alloc] initWithStringEncoding:encoding]; return self; } - (void)dealloc { [bodyStream release]; [super dealloc]; } - (void)appendPartWithFormData:(NSData *)data name:(NSString *)name { [bodyStream addFormData:data name:name]; } - (BOOL)appendPartWithFileURL:(NSURL *)fileURL name:(NSString *)name error:(NSError **)error { return [bodyStream addFileFromURL:fileURL name:name error:error]; } - (NSMutableURLRequest *)requestByFinalizingMultipartFormData { //return the original request if no data has been added if ([bodyStream empty]) { return self.request; } [self.request setValue:[NSString stringWithFormat:@"multipart/form-data; boundary=%@", kAFMultipartFormBoundary] forHTTPHeaderField:@"Content-Type"]; [self.request setValue:[NSString stringWithFormat:@"%d",[bodyStream contentLength]] forHTTPHeaderField:@"Content-Length"]; [self.request setHTTPBodyStream:bodyStream]; return self.request; } - (void)appendData:(NSData *)data { } @end #pragma mark - AFMultipartBodyStream @interface AFMultipartBodyStream () { CFReadStreamClientCallBack copiedCallback; CFStreamClientContext copiedContext; CFOptionFlags requestedEvents; NSStreamStatus streamStatus; id delegate; NSMutableArray *fileNames; NSMutableDictionary *fileURLs; NSMutableDictionary *fileHeaders; NSMutableArray *formNames; NSMutableDictionary *formDatas; NSMutableDictionary *formHeaders; NSUInteger readElementCursor; NSUInteger readOffsetCursor; NSUInteger readHeaderOffsetCursor; NSInputStream * currentFileStream; NSStringEncoding stringEncoding; } @end @implementation AFMultipartBodyStream - (id)init { self = [super init]; fileNames = [[NSMutableArray alloc] init]; fileURLs = [[NSMutableDictionary alloc] init]; fileHeaders = [[NSMutableDictionary alloc] init]; formNames = [[NSMutableArray alloc] init]; formDatas = [[NSMutableDictionary alloc] init]; formHeaders = [[NSMutableDictionary alloc] init]; currentFileStream = NULL; stringEncoding = NSUTF8StringEncoding; streamStatus = NSStreamStatusNotOpen; [self resetCursors]; // [self setDelegate:self]; return self; } - (id)initWithStringEncoding:(NSStringEncoding)encoding { self = [self init]; stringEncoding = encoding; return self; } - (void)resetCursors { readElementCursor = 0; readOffsetCursor = 0; readHeaderOffsetCursor = 0; } - (void)dealloc { [fileNames release]; [fileURLs release]; [fileHeaders release]; [formNames release]; [formDatas release]; [formHeaders release]; if (currentFileStream) { [currentFileStream close]; [currentFileStream release]; currentFileStream = NULL; } [super dealloc]; } - (BOOL)empty { if (([fileURLs count] + [formDatas count]) == 0) return YES; else return NO; } - (BOOL)addFileFromURL:(NSURL *)fileURL name:(NSString *)name error:(NSError **)error { assert([self streamStatus] == NSStreamStatusNotOpen); NSMutableDictionary *userInfo = [NSMutableDictionary dictionary]; [userInfo setValue:fileURL forKey:NSURLErrorFailingURLErrorKey]; if (![fileURL isFileURL]) { [userInfo setValue:NSLocalizedString(@"Expected URL to be a file URL", nil) forKey:NSLocalizedFailureReasonErrorKey]; if (error != NULL) { *error = [[[NSError alloc] initWithDomain:AFNetworkingErrorDomain code:NSURLErrorBadURL userInfo:userInfo] autorelease]; } return NO; } if ([fileURL checkResourceIsReachableAndReturnError:error] == NO) { [userInfo setValue:NSLocalizedString(@"File URL not reachable.", nil) forKey:NSLocalizedFailureReasonErrorKey]; if (error != NULL) { *error = [[[NSError alloc] initWithDomain:AFNetworkingErrorDomain code:NSURLErrorBadURL userInfo:userInfo] autorelease]; } return NO; } [fileNames addObject:name]; [fileURLs setObject:fileURL forKey:name]; [self generateHeaders]; return YES; } - (void)addFormData:(NSData *)data name:(NSString *)name { assert([self streamStatus] == NSStreamStatusNotOpen); [formNames addObject:name]; [formDatas setObject:data forKey:name]; [self generateHeaders]; } - (void)generateHeaders { [formHeaders removeAllObjects]; for (NSString * formName in formNames) { [formHeaders setObject:[self headersDataForForm:formName] forKey:formName]; } [fileHeaders removeAllObjects]; for (NSString * fileName in fileNames) { [fileHeaders setObject:[self headersDataForFile:fileName] forKey:fileName]; } } - (NSString *)stringForHeaders:(NSDictionary *)headers { NSMutableString * headerString = [NSMutableString string]; for (NSString *field in [headers allKeys]) { [headerString appendString:[NSString stringWithFormat:@"%@: %@%@", field, [headers valueForKey:field], kAFMultipartFormCRLF]]; } return [NSString stringWithString:headerString]; } - (NSData *)headersDataForDict:(NSDictionary *)headersDict { NSMutableString * headerString = [NSMutableString string]; if ([formHeaders count] == 0 && [fileHeaders count] == 0) { [headerString appendString:AFMultipartFormInitialBoundary()]; } else { [headerString appendString:AFMultipartFormEncapsulationBoundary()]; } [headerString appendString:[self stringForHeaders:headersDict]]; [headerString appendString:kAFMultipartFormCRLF]; return [headerString dataUsingEncoding:stringEncoding]; } - (NSData *)headersDataForFile:(NSString *)name { NSURL * fileURL = [fileURLs objectForKey:name]; NSString * fileName = [[fileURL pathComponents] objectAtIndex:([[fileURL pathComponents] count] - 1)]; NSMutableDictionary * mutableHeaders = [NSMutableDictionary dictionary]; [mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"; filename=\"%@\"", name, fileName] forKey:@"Content-Disposition"]; CFStringRef UTI = UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, (CFStringRef)[fileURL pathExtension], NULL); NSString * MIMEType = [(NSString*) UTTypeCopyPreferredTagWithClass (UTI, kUTTagClassMIMEType) autorelease]; CFRelease(UTI); [mutableHeaders setValue:MIMEType forKey:@"Content-Type"]; return [self headersDataForDict:mutableHeaders]; } - (NSData *)headersDataForForm:(NSString *)name { return [self headersDataForDict:[NSDictionary dictionaryWithObject:[NSString stringWithFormat:@"form-data; name=\"%@\"", name] forKey:@"Content-Disposition"]]; } - (NSInteger)readData:(NSData *)data intoBuffer:(uint8_t *)buffer maxLength:(NSUInteger)len offsetCursor:(NSUInteger*)offsetCursorPtr { NSInteger bytesAvailable = [data length] - *offsetCursorPtr; if (len > bytesAvailable) { [data getBytes:buffer range:NSMakeRange(*offsetCursorPtr, bytesAvailable)]; *offsetCursorPtr += bytesAvailable; return bytesAvailable; } else { [data getBytes:buffer range:NSMakeRange(*offsetCursorPtr, len)]; *offsetCursorPtr += len; return len; } } - (void)nextElement { readOffsetCursor = 0; readHeaderOffsetCursor = 0; readElementCursor += 1; if (currentFileStream) { [currentFileStream close]; [currentFileStream release]; currentFileStream = NULL; } } - (NSUInteger)contentLength { NSUInteger total = 0; for (NSString * formName in formNames) { total += [[formHeaders objectForKey:formName] length]; total += [[formDatas objectForKey:formName] length]; } for (NSString * fileName in fileNames) { NSError * error; NSDictionary *fileAttributes = [[NSFileManager defaultManager] attributesOfItemAtPath:[[fileURLs objectForKey:fileName] path] error:&error]; total += [[fileHeaders objectForKey:fileName] length]; total += [[fileAttributes objectForKey:NSFileSize] longValue]; } total += [[self finalBoundaryData] length]; return total; } - (NSUInteger)totalElements { return [formDatas count] + [fileURLs count] + 1; //one extra for final boundary } - (NSData *)finalBoundaryData { return [AFMultipartFormFinalBoundary() dataUsingEncoding:stringEncoding]; } #pragma mark - NSStream subclass overrides - (void)open { streamStatus = NSStreamStatusOpen; } - (void)close { streamStatus = NSStreamStatusClosed; [self resetCursors]; } - (id )delegate { return delegate; } - (void)setDelegate:(id)aDelegate { if (aDelegate == nil) { delegate = self; } else { delegate = aDelegate; } } - (void)scheduleInRunLoop:(NSRunLoop *)aRunLoop forMode:(NSString *)mode { } - (void)removeFromRunLoop:(NSRunLoop *)aRunLoop forMode:(NSString *)mode { } - (id)propertyForKey:(NSString *)key { return nil; } - (BOOL)setProperty:(id)property forKey:(NSString *)key { return NO; } - (NSStreamStatus)streamStatus { return streamStatus; } - (NSError *)streamError { return nil; } #pragma mark - NSInputStream subclass overrides - (NSInteger)read:(uint8_t *)buffer maxLength:(NSUInteger)len { if ([self streamStatus] == NSStreamStatusClosed) { return 0; } assert ([self streamStatus] == NSStreamStatusOpen); NSInteger bytesRead = 0; NSInteger readFileCursor = (readElementCursor - [formNames count]); if (readElementCursor < [formNames count]) { //reading from formDatas NSString * currentFormName = [formNames objectAtIndex:readElementCursor]; NSData * currentData = [formDatas objectForKey:currentFormName]; NSData * headersData = [formHeaders objectForKey:currentFormName]; if (readHeaderOffsetCursor < [headersData length]) { bytesRead = [self readData:headersData intoBuffer:buffer maxLength:len offsetCursor:&readHeaderOffsetCursor]; } else { bytesRead = [self readData:currentData intoBuffer:buffer maxLength:len offsetCursor:&readOffsetCursor]; } if (readOffsetCursor == [currentData length]) { [self nextElement]; } } else if (readFileCursor >= 0 && readFileCursor < [fileNames count]) { //reading from files NSString * currentFileName = [fileNames objectAtIndex:readFileCursor]; NSURL * currentFileURL = [fileURLs objectForKey:currentFileName]; NSData * headersData = [fileHeaders objectForKey:currentFileName]; assert(headersData != NULL); if (readHeaderOffsetCursor < [headersData length]) { bytesRead = [self readData:headersData intoBuffer:buffer maxLength:len offsetCursor:&readHeaderOffsetCursor]; } else { if (!currentFileStream) { currentFileStream = [[NSInputStream inputStreamWithURL:currentFileURL] retain]; // currentFileStream.delegate = self; [currentFileStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSRunLoopCommonModes]; [currentFileStream open]; } bytesRead = [currentFileStream read:buffer maxLength:len]; if (![currentFileStream hasBytesAvailable] || bytesRead == 0) { [self nextElement]; } } } else if (readElementCursor < [self totalElements]) { //add final boundary bytesRead = [self readData:[self finalBoundaryData] intoBuffer:buffer maxLength:len offsetCursor:&readOffsetCursor]; if (readOffsetCursor == [[self finalBoundaryData] length]) { [self nextElement]; } } else { [self nextElement]; } if (readElementCursor <= [self totalElements]) { if (bytesRead < len) { bytesRead += [self read:buffer+bytesRead maxLength:len-bytesRead]; } else { // no deeper recursion so call callback if necessary if (copiedCallback && (requestedEvents & kCFStreamEventHasBytesAvailable)) { copiedCallback((CFReadStreamRef)self, kCFStreamEventHasBytesAvailable, &copiedContext); } } } return bytesRead; } - (BOOL)hasBytesAvailable { if ([self streamStatus] != NSStreamStatusOpen) { return NO; } else { return YES; } } - (BOOL)getBuffer:(uint8_t **)buffer length:(NSUInteger *)len { return NO; } #pragma mark - Undocumented CFReadStream bridged methods - (void)_scheduleInCFRunLoop:(CFRunLoopRef)aRunLoop forMode:(CFStringRef)aMode { } - (BOOL)_setCFClientFlags:(CFOptionFlags)inFlags callback:(CFReadStreamClientCallBack)inCallback context:(CFStreamClientContext *)inContext { if (inCallback != NULL) { requestedEvents = inFlags; copiedCallback = inCallback; memcpy(&copiedContext, inContext, sizeof(CFStreamClientContext)); if (copiedContext.info && copiedContext.retain) { copiedContext.retain(copiedContext.info); } copiedCallback((CFReadStreamRef)self, kCFStreamEventHasBytesAvailable, &copiedContext); } else { requestedEvents = kCFStreamEventNone; copiedCallback = NULL; if (copiedContext.info && copiedContext.release) { copiedContext.release(copiedContext.info); } memset(&copiedContext, 0, sizeof(CFStreamClientContext)); } return YES; } - (void)_unscheduleFromCFRunLoop:(CFRunLoopRef)aRunLoop forMode:(CFStringRef)aMode { } @end