a610a89f8c
Ran into some issues with sending an unescaped apostrophe, but by adding it to the list of characters to escape, this problem is now fixed for me.
1208 lines
47 KiB
Objective-C
1208 lines
47 KiB
Objective-C
// 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 <Foundation/Foundation.h>
|
|
|
|
#import "AFHTTPClient.h"
|
|
#import "AFHTTPRequestOperation.h"
|
|
|
|
#import <Availability.h>
|
|
|
|
#ifdef _SYSTEMCONFIGURATION_H
|
|
#import <netinet/in.h>
|
|
#import <netinet6/in6.h>
|
|
#import <arpa/inet.h>
|
|
#import <ifaddrs.h>
|
|
#import <netdb.h>
|
|
#endif
|
|
|
|
#if defined(__IPHONE_OS_VERSION_MIN_REQUIRED)
|
|
#import <UIKit/UIKit.h>
|
|
#endif
|
|
|
|
#ifdef _SYSTEMCONFIGURATION_H
|
|
NSString * const AFNetworkingReachabilityDidChangeNotification = @"com.alamofire.networking.reachability.change";
|
|
NSString * const AFNetworkingReachabilityNotificationStatusItem = @"AFNetworkingReachabilityNotificationStatusItem";
|
|
|
|
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];
|
|
}
|
|
|
|
static NSString * AFPercentEscapedQueryStringPairMemberFromStringWithEncoding(NSString *string, NSStringEncoding encoding) {
|
|
static NSString * const kAFCharactersToBeEscaped = @":/?&=;+!@#$()~'";
|
|
static NSString * const kAFCharactersToLeaveUnescaped = @"[].";
|
|
|
|
return (__bridge_transfer NSString *)CFURLCreateStringByAddingPercentEscapes(kCFAllocatorDefault, (__bridge CFStringRef)string, (__bridge CFStringRef)kAFCharactersToLeaveUnescaped, (__bridge CFStringRef)kAFCharactersToBeEscaped, CFStringConvertNSStringEncodingToEncoding(encoding));
|
|
}
|
|
|
|
#pragma mark -
|
|
|
|
@interface AFQueryStringPair : NSObject
|
|
@property (readwrite, nonatomic, strong) id field;
|
|
@property (readwrite, nonatomic, strong) id value;
|
|
|
|
- (id)initWithField:(id)field value:(id)value;
|
|
|
|
- (NSString *)URLEncodedStringValueWithEncoding:(NSStringEncoding)stringEncoding;
|
|
@end
|
|
|
|
@implementation AFQueryStringPair
|
|
@synthesize field = _field;
|
|
@synthesize value = _value;
|
|
|
|
- (id)initWithField:(id)field value:(id)value {
|
|
self = [super init];
|
|
if (!self) {
|
|
return nil;
|
|
}
|
|
|
|
self.field = field;
|
|
self.value = value;
|
|
|
|
return self;
|
|
}
|
|
|
|
- (NSString *)URLEncodedStringValueWithEncoding:(NSStringEncoding)stringEncoding {
|
|
if (!self.value || [self.value isEqual:[NSNull null]]) {
|
|
return AFPercentEscapedQueryStringPairMemberFromStringWithEncoding([self.field description], stringEncoding);
|
|
} else {
|
|
return [NSString stringWithFormat:@"%@=%@", AFPercentEscapedQueryStringPairMemberFromStringWithEncoding([self.field description], stringEncoding), AFPercentEscapedQueryStringPairMemberFromStringWithEncoding([self.value description], stringEncoding)];
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
extern NSArray * AFQueryStringPairsFromDictionary(NSDictionary *dictionary);
|
|
extern NSArray * AFQueryStringPairsFromKeyAndValue(NSString *key, id value);
|
|
|
|
NSString * AFQueryStringFromParametersWithEncoding(NSDictionary *parameters, NSStringEncoding stringEncoding) {
|
|
NSMutableArray *mutablePairs = [NSMutableArray array];
|
|
for (AFQueryStringPair *pair in AFQueryStringPairsFromDictionary(parameters)) {
|
|
[mutablePairs addObject:[pair URLEncodedStringValueWithEncoding:stringEncoding]];
|
|
}
|
|
|
|
return [mutablePairs componentsJoinedByString:@"&"];
|
|
}
|
|
|
|
NSArray * AFQueryStringPairsFromDictionary(NSDictionary *dictionary) {
|
|
return AFQueryStringPairsFromKeyAndValue(nil, dictionary);
|
|
}
|
|
|
|
NSArray * AFQueryStringPairsFromKeyAndValue(NSString *key, id value) {
|
|
NSMutableArray *mutableQueryStringComponents = [NSMutableArray array];
|
|
|
|
if ([value isKindOfClass:[NSDictionary class]]) {
|
|
// Sort dictionary keys to ensure consistent ordering in query string, which is important when deserializing potentially ambiguous sequences, such as an array of dictionaries
|
|
NSSortDescriptor *sortDescriptor = [NSSortDescriptor sortDescriptorWithKey:@"description" ascending:YES selector:@selector(caseInsensitiveCompare:)];
|
|
[[[value allKeys] sortedArrayUsingDescriptors:[NSArray arrayWithObject:sortDescriptor]] enumerateObjectsUsingBlock:^(id nestedKey, __unused NSUInteger idx, __unused BOOL *stop) {
|
|
id nestedValue = [value objectForKey:nestedKey];
|
|
if (nestedValue) {
|
|
[mutableQueryStringComponents addObjectsFromArray:AFQueryStringPairsFromKeyAndValue((key ? [NSString stringWithFormat:@"%@[%@]", key, nestedKey] : nestedKey), nestedValue)];
|
|
}
|
|
}];
|
|
} else if ([value isKindOfClass:[NSArray class]]) {
|
|
NSArray *array = value;
|
|
[array enumerateObjectsUsingBlock:^(id nestedValue, __unused NSUInteger idx, __unused BOOL *stop) {
|
|
[mutableQueryStringComponents addObjectsFromArray:AFQueryStringPairsFromKeyAndValue([NSString stringWithFormat:@"%@[]", key], nestedValue)];
|
|
}];
|
|
} else {
|
|
[mutableQueryStringComponents addObject:[[AFQueryStringPair alloc] initWithField:key value:value]];
|
|
}
|
|
|
|
return mutableQueryStringComponents;
|
|
}
|
|
|
|
@interface AFStreamingMultipartFormData : NSObject <AFMultipartFormData>
|
|
- (id)initWithURLRequest:(NSMutableURLRequest *)urlRequest
|
|
stringEncoding:(NSStringEncoding)encoding;
|
|
|
|
- (NSMutableURLRequest *)requestByFinalizingMultipartFormData;
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@interface AFHTTPClient ()
|
|
@property (readwrite, nonatomic) NSURL *baseURL;
|
|
@property (readwrite, nonatomic) NSMutableArray *registeredHTTPOperationClassNames;
|
|
@property (readwrite, nonatomic) NSMutableDictionary *defaultHeaders;
|
|
@property (readwrite, nonatomic) 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
|
|
|
|
+ (instancetype)clientWithBaseURL:(NSURL *)url {
|
|
return [[self alloc] initWithBaseURL:url];
|
|
}
|
|
|
|
- (id)initWithBaseURL:(NSURL *)url {
|
|
NSParameterAssert(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-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 defined(__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:@"%@/%@ (%@; iOS %@; Scale/%0.2f)", [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleExecutableKey] ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleIdentifierKey], (__bridge id)CFBundleGetValueForInfoDictionaryKey(CFBundleGetMainBundle(), kCFBundleVersionKey) ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleVersionKey], [[UIDevice currentDevice] model], [[UIDevice currentDevice] systemVersion], ([[UIScreen mainScreen] respondsToSelector:@selector(scale)] ? [[UIScreen mainScreen] scale] : 1.0f)]];
|
|
#elif defined(__MAC_OS_X_VERSION_MIN_REQUIRED)
|
|
[self setDefaultHeader:@"User-Agent" value:[NSString stringWithFormat:@"%@/%@ (Mac OS X %@)", [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleExecutableKey] ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleIdentifierKey], [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleShortVersionString"] ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(NSString *)kCFBundleVersionKey], [[NSProcessInfo processInfo] operatingSystemVersionString]]];
|
|
#endif
|
|
|
|
#ifdef _SYSTEMCONFIGURATION_H
|
|
self.networkReachabilityStatus = AFNetworkReachabilityStatusUnknown;
|
|
[self startMonitoringNetworkReachability];
|
|
#endif
|
|
|
|
self.operationQueue = [[NSOperationQueue alloc] init];
|
|
[self.operationQueue setMaxConcurrentOperationCount:NSOperationQueueDefaultMaxConcurrentOperationCount];
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc {
|
|
#ifdef _SYSTEMCONFIGURATION_H
|
|
[self stopMonitoringNetworkReachability];
|
|
#endif
|
|
}
|
|
|
|
- (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 = (__bridge AFNetworkReachabilityStatusBlock)info;
|
|
if (block) {
|
|
block(status);
|
|
}
|
|
|
|
dispatch_async(dispatch_get_main_queue(), ^{
|
|
[[NSNotificationCenter defaultCenter] postNotificationName:AFNetworkingReachabilityDidChangeNotification object:nil userInfo:[NSDictionary dictionaryWithObject:[NSNumber numberWithInteger:status] forKey:AFNetworkingReachabilityNotificationStatusItem]];
|
|
});
|
|
}
|
|
|
|
static const void * AFNetworkReachabilityRetainCallback(const void *info) {
|
|
return (__bridge_retained const void *)([(__bridge AFNetworkReachabilityStatusBlock)info copy]);
|
|
}
|
|
|
|
static void AFNetworkReachabilityReleaseCallback(const void *info) {
|
|
if (info) {
|
|
CFRelease(info);
|
|
}
|
|
}
|
|
|
|
- (void)startMonitoringNetworkReachability {
|
|
[self stopMonitoringNetworkReachability];
|
|
|
|
if (!self.baseURL) {
|
|
return;
|
|
}
|
|
|
|
self.networkReachability = SCNetworkReachabilityCreateWithName(kCFAllocatorDefault, [[self.baseURL host] UTF8String]);
|
|
|
|
if (!self.networkReachability) {
|
|
return;
|
|
}
|
|
|
|
__weak __typeof(&*self)weakSelf = self;
|
|
AFNetworkReachabilityStatusBlock callback = ^(AFNetworkReachabilityStatus status) {
|
|
__strong __typeof(&*weakSelf)strongSelf = weakSelf;
|
|
if (!strongSelf) {
|
|
return;
|
|
}
|
|
|
|
strongSelf.networkReachabilityStatus = status;
|
|
if (strongSelf.networkReachabilityStatusBlock) {
|
|
strongSelf.networkReachabilityStatusBlock(status);
|
|
}
|
|
};
|
|
|
|
SCNetworkReachabilityContext context = {0, (__bridge void *)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);
|
|
_networkReachability = NULL;
|
|
}
|
|
}
|
|
|
|
- (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
|
|
{
|
|
NSParameterAssert(method);
|
|
|
|
if (!path) {
|
|
path = @"";
|
|
}
|
|
|
|
NSURL *url = [NSURL URLWithString:path relativeToURL:self.baseURL];
|
|
NSMutableURLRequest *request = [[NSMutableURLRequest alloc] initWithURL:url];
|
|
[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 = (__bridge NSString *)CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(self.stringEncoding));
|
|
NSError *error = nil;
|
|
|
|
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:[NSJSONSerialization dataWithJSONObject:parameters options:0 error:&error]];
|
|
break;
|
|
case AFPropertyListParameterEncoding:;
|
|
[request setValue:[NSString stringWithFormat:@"application/x-plist; charset=%@", charset] forHTTPHeaderField:@"Content-Type"];
|
|
[request setHTTPBody:[NSPropertyListSerialization dataWithPropertyList:parameters format:NSPropertyListXMLFormat_v1_0 options:0 error:&error]];
|
|
break;
|
|
}
|
|
|
|
if (error) {
|
|
NSLog(@"%@ %@: %@", [self class], NSStringFromSelector(_cmd), error);
|
|
}
|
|
}
|
|
}
|
|
|
|
return request;
|
|
}
|
|
|
|
- (NSMutableURLRequest *)multipartFormRequestWithMethod:(NSString *)method
|
|
path:(NSString *)path
|
|
parameters:(NSDictionary *)parameters
|
|
constructingBodyWithBlock:(void (^)(id <AFMultipartFormData> formData))block
|
|
{
|
|
NSParameterAssert(method);
|
|
NSParameterAssert(![method isEqualToString:@"GET"] && ![method isEqualToString:@"HEAD"]);
|
|
|
|
NSMutableURLRequest *request = [self requestWithMethod:method path:path parameters:nil];
|
|
|
|
__block AFStreamingMultipartFormData *formData = [[AFStreamingMultipartFormData alloc] initWithURLRequest:request stringEncoding:self.stringEncoding];
|
|
|
|
if (parameters) {
|
|
for (AFQueryStringPair *pair in AFQueryStringPairsFromDictionary(parameters)) {
|
|
NSData *data = nil;
|
|
if ([pair.value isKindOfClass:[NSData class]]) {
|
|
data = pair.value;
|
|
} else if ([pair.value isEqual:[NSNull null]]) {
|
|
data = [NSData data];
|
|
} else {
|
|
data = [[pair.value description] dataUsingEncoding:self.stringEncoding];
|
|
}
|
|
|
|
if (data) {
|
|
[formData appendPartWithFormData:data name:[pair.field 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];
|
|
}
|
|
}
|
|
|
|
if (!operation) {
|
|
operation = [[AFHTTPRequestOperation alloc] initWithRequest:urlRequest];
|
|
}
|
|
|
|
[operation setCompletionBlockWithSuccess:success failure:failure];
|
|
|
|
return operation;
|
|
}
|
|
|
|
#pragma mark -
|
|
|
|
- (void)enqueueHTTPRequestOperation:(AFHTTPRequestOperation *)operation {
|
|
[self.operationQueue addOperation:operation];
|
|
}
|
|
|
|
- (void)cancelAllHTTPOperationsWithMethod:(NSString *)method
|
|
path:(NSString *)path
|
|
{
|
|
NSString *URLStringToMatched = [[[self requestWithMethod:(method ?: @"GET") path:path parameters:nil] URL] absoluteString];
|
|
|
|
for (NSOperation *operation in [self.operationQueue operations]) {
|
|
if (![operation isKindOfClass:[AFHTTPRequestOperation class]]) {
|
|
continue;
|
|
}
|
|
|
|
BOOL hasMatchingMethod = !method || [method isEqualToString:[[(AFHTTPRequestOperation *)operation request] HTTPMethod]];
|
|
BOOL hasMatchingURL = [[[[(AFHTTPRequestOperation *)operation request] URL] absoluteString] isEqualToString:URLStringToMatched];
|
|
|
|
if (hasMatchingMethod && hasMatchingURL) {
|
|
[operation cancel];
|
|
}
|
|
}
|
|
}
|
|
|
|
- (void)enqueueBatchOfHTTPRequestOperationsWithRequests:(NSArray *)urlRequests
|
|
progressBlock:(void (^)(NSUInteger numberOfFinishedOperations, NSUInteger totalNumberOfOperations))progressBlock
|
|
completionBlock:(void (^)(NSArray *operations))completionBlock
|
|
{
|
|
NSMutableArray *mutableOperations = [NSMutableArray array];
|
|
for (NSURLRequest *request in urlRequests) {
|
|
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 numberOfFinishedOperations, 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);
|
|
}
|
|
});
|
|
#if !OS_OBJECT_USE_OBJC
|
|
dispatch_release(dispatchGroup);
|
|
#endif
|
|
}];
|
|
|
|
for (AFHTTPRequestOperation *operation in operations) {
|
|
AFCompletionBlock originalCompletionBlock = [operation.completionBlock copy];
|
|
__weak AFHTTPRequestOperation *weakOperation = operation;
|
|
operation.completionBlock = ^{
|
|
dispatch_queue_t queue = weakOperation.successCallbackQueue ?: dispatch_get_main_queue();
|
|
dispatch_group_async(dispatchGroup, queue, ^{
|
|
if (originalCompletionBlock) {
|
|
originalCompletionBlock();
|
|
}
|
|
|
|
__block NSUInteger numberOfFinishedOperations = 0;
|
|
[operations enumerateObjectsUsingBlock:^(id obj, __unused NSUInteger idx, __unused BOOL *stop) {
|
|
if ([(NSOperation *)obj isFinished]) {
|
|
numberOfFinishedOperations++;
|
|
}
|
|
}];
|
|
|
|
if (progressBlock) {
|
|
progressBlock(numberOfFinishedOperations, [operations count]);
|
|
}
|
|
|
|
dispatch_group_leave(dispatchGroup);
|
|
});
|
|
};
|
|
|
|
dispatch_group_enter(dispatchGroup);
|
|
[batchedOperation addDependency:operation];
|
|
}
|
|
[self.operationQueue addOperations:operations waitUntilFinished:NO];
|
|
[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:(NSInteger)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];
|
|
HTTPClient.defaultHeaders = [self.defaultHeaders copyWithZone:zone];
|
|
#ifdef _SYSTEMCONFIGURATION_H
|
|
HTTPClient.networkReachabilityStatusBlock = self.networkReachabilityStatusBlock;
|
|
#endif
|
|
return HTTPClient;
|
|
}
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
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];
|
|
}
|
|
|
|
static inline NSString * AFContentTypeForPathExtension(NSString *extension) {
|
|
#ifdef __UTTYPE__
|
|
NSString *UTI = (__bridge_transfer NSString *)UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, (__bridge CFStringRef)extension, NULL);
|
|
return (__bridge_transfer NSString *)UTTypeCopyPreferredTagWithClass((__bridge CFStringRef)UTI, kUTTagClassMIMEType);
|
|
#else
|
|
return @"application/octet-stream";
|
|
#endif
|
|
}
|
|
|
|
NSUInteger const kAFUploadStream3GSuggestedPacketSize = 1024 * 16;
|
|
NSTimeInterval const kAFUploadStream3GSuggestedDelay = 0.2;
|
|
|
|
@interface AFHTTPBodyPart : NSObject
|
|
@property (nonatomic, assign) NSStringEncoding stringEncoding;
|
|
@property (nonatomic, strong) NSDictionary *headers;
|
|
@property (nonatomic, strong) NSInputStream *inputStream;
|
|
@property (nonatomic, assign) unsigned long long bodyContentLength;
|
|
|
|
@property (nonatomic, assign) BOOL hasInitialBoundary;
|
|
@property (nonatomic, assign) BOOL hasFinalBoundary;
|
|
|
|
@property (nonatomic, readonly, getter = hasBytesAvailable) BOOL bytesAvailable;
|
|
@property (nonatomic, readonly) unsigned long long contentLength;
|
|
|
|
- (NSInteger)read:(uint8_t *)buffer maxLength:(NSUInteger)length;
|
|
@end
|
|
|
|
@interface AFMultipartBodyStream : NSInputStream <NSStreamDelegate>
|
|
@property (nonatomic, assign) NSUInteger numberOfBytesInPacket;
|
|
@property (nonatomic, assign) NSTimeInterval delay;
|
|
@property (nonatomic, readonly) unsigned long long contentLength;
|
|
@property (nonatomic, readonly, getter = isEmpty) BOOL empty;
|
|
|
|
- (id)initWithStringEncoding:(NSStringEncoding)encoding;
|
|
- (void)setInitialAndFinalBoundaries;
|
|
- (void)appendHTTPBodyPart:(AFHTTPBodyPart *)bodyPart;
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@interface AFStreamingMultipartFormData ()
|
|
@property (readwrite, nonatomic, copy) NSMutableURLRequest *request;
|
|
@property (readwrite, nonatomic, strong) AFMultipartBodyStream *bodyStream;
|
|
@property (readwrite, nonatomic, assign) NSStringEncoding stringEncoding;
|
|
@end
|
|
|
|
@implementation AFStreamingMultipartFormData
|
|
@synthesize request = _request;
|
|
@synthesize bodyStream = _bodyStream;
|
|
@synthesize stringEncoding = _stringEncoding;
|
|
|
|
- (id)initWithURLRequest:(NSMutableURLRequest *)urlRequest
|
|
stringEncoding:(NSStringEncoding)encoding
|
|
{
|
|
self = [super init];
|
|
if (!self) {
|
|
return nil;
|
|
}
|
|
|
|
self.request = urlRequest;
|
|
self.stringEncoding = encoding;
|
|
self.bodyStream = [[AFMultipartBodyStream alloc] initWithStringEncoding:encoding];
|
|
|
|
return self;
|
|
}
|
|
|
|
- (BOOL)appendPartWithFileURL:(NSURL *)fileURL
|
|
name:(NSString *)name
|
|
error:(NSError * __autoreleasing *)error
|
|
{
|
|
NSParameterAssert(fileURL);
|
|
NSParameterAssert(name);
|
|
|
|
if (![fileURL isFileURL]) {
|
|
NSDictionary *userInfo = [NSDictionary dictionaryWithObject:NSLocalizedStringFromTable(@"Expected URL to be a file URL", @"AFNetworking", nil) forKey:NSLocalizedFailureReasonErrorKey];
|
|
if (error != NULL) {
|
|
*error = [[NSError alloc] initWithDomain:AFNetworkingErrorDomain code:NSURLErrorBadURL userInfo:userInfo];
|
|
}
|
|
|
|
return NO;
|
|
} else if ([fileURL checkResourceIsReachableAndReturnError:error] == NO) {
|
|
NSDictionary *userInfo = [NSDictionary dictionaryWithObject:NSLocalizedStringFromTable(@"File URL not reachable.", @"AFNetworking", nil) forKey:NSLocalizedFailureReasonErrorKey];
|
|
if (error != NULL) {
|
|
*error = [[NSError alloc] initWithDomain:AFNetworkingErrorDomain code:NSURLErrorBadURL userInfo:userInfo];
|
|
}
|
|
|
|
return NO;
|
|
}
|
|
|
|
NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
|
|
[mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"; filename=\"%@\"", name, [fileURL lastPathComponent]] forKey:@"Content-Disposition"];
|
|
[mutableHeaders setValue:AFContentTypeForPathExtension([fileURL pathExtension]) forKey:@"Content-Type"];
|
|
|
|
AFHTTPBodyPart *bodyPart = [[AFHTTPBodyPart alloc] init];
|
|
bodyPart.stringEncoding = self.stringEncoding;
|
|
bodyPart.headers = mutableHeaders;
|
|
bodyPart.inputStream = [NSInputStream inputStreamWithURL:fileURL];
|
|
|
|
NSDictionary *fileAttributes = [[NSFileManager defaultManager] attributesOfItemAtPath:[fileURL path] error:nil];
|
|
bodyPart.bodyContentLength = [[fileAttributes objectForKey:NSFileSize] unsignedLongLongValue];
|
|
|
|
[self.bodyStream appendHTTPBodyPart:bodyPart];
|
|
|
|
return YES;
|
|
}
|
|
|
|
- (void)appendPartWithFileData:(NSData *)data
|
|
name:(NSString *)name
|
|
fileName:(NSString *)fileName
|
|
mimeType:(NSString *)mimeType
|
|
{
|
|
NSParameterAssert(name);
|
|
NSParameterAssert(fileName);
|
|
NSParameterAssert(mimeType);
|
|
|
|
NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
|
|
[mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"; filename=\"%@\"", name, fileName] forKey:@"Content-Disposition"];
|
|
[mutableHeaders setValue:mimeType forKey:@"Content-Type"];
|
|
|
|
[self appendPartWithHeaders:mutableHeaders body:data];
|
|
}
|
|
|
|
- (void)appendPartWithFormData:(NSData *)data
|
|
name:(NSString *)name
|
|
{
|
|
NSParameterAssert(name);
|
|
|
|
NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
|
|
[mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"", name] forKey:@"Content-Disposition"];
|
|
|
|
[self appendPartWithHeaders:mutableHeaders body:data];
|
|
}
|
|
|
|
- (void)appendPartWithHeaders:(NSDictionary *)headers
|
|
body:(NSData *)body
|
|
{
|
|
NSParameterAssert(body);
|
|
|
|
AFHTTPBodyPart *bodyPart = [[AFHTTPBodyPart alloc] init];
|
|
bodyPart.stringEncoding = self.stringEncoding;
|
|
bodyPart.headers = headers;
|
|
bodyPart.bodyContentLength = [body length];
|
|
bodyPart.inputStream = [NSInputStream inputStreamWithData:body];
|
|
|
|
[self.bodyStream appendHTTPBodyPart:bodyPart];
|
|
}
|
|
|
|
- (void)throttleBandwidthWithPacketSize:(NSUInteger)numberOfBytes
|
|
delay:(NSTimeInterval)delay
|
|
{
|
|
self.bodyStream.numberOfBytesInPacket = numberOfBytes;
|
|
self.bodyStream.delay = delay;
|
|
}
|
|
|
|
- (NSMutableURLRequest *)requestByFinalizingMultipartFormData {
|
|
if ([self.bodyStream isEmpty]) {
|
|
return self.request;
|
|
}
|
|
|
|
// Reset the initial and final boundaries to ensure correct Content-Length
|
|
[self.bodyStream setInitialAndFinalBoundaries];
|
|
|
|
[self.request setValue:[NSString stringWithFormat:@"multipart/form-data; boundary=%@", kAFMultipartFormBoundary] forHTTPHeaderField:@"Content-Type"];
|
|
[self.request setValue:[NSString stringWithFormat:@"%llu", [self.bodyStream contentLength]] forHTTPHeaderField:@"Content-Length"];
|
|
[self.request setHTTPBodyStream:self.bodyStream];
|
|
|
|
return self.request;
|
|
}
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@interface AFMultipartBodyStream ()
|
|
@property (nonatomic, assign) NSStreamStatus streamStatus;
|
|
@property (nonatomic, strong) NSError *streamError;
|
|
|
|
@property (nonatomic, assign) NSStringEncoding stringEncoding;
|
|
@property (nonatomic, strong) NSMutableArray *HTTPBodyParts;
|
|
@property (nonatomic, strong) NSEnumerator *HTTPBodyPartEnumerator;
|
|
@property (nonatomic, strong) AFHTTPBodyPart *currentHTTPBodyPart;
|
|
@end
|
|
|
|
@implementation AFMultipartBodyStream
|
|
@synthesize streamStatus = _streamStatus;
|
|
@synthesize streamError = _streamError;
|
|
@synthesize stringEncoding = _stringEncoding;
|
|
@synthesize HTTPBodyParts = _HTTPBodyParts;
|
|
@synthesize HTTPBodyPartEnumerator = _HTTPBodyPartEnumerator;
|
|
@synthesize currentHTTPBodyPart = _currentHTTPBodyPart;
|
|
@synthesize numberOfBytesInPacket = _numberOfBytesInPacket;
|
|
@synthesize delay = _delay;
|
|
|
|
- (id)initWithStringEncoding:(NSStringEncoding)encoding {
|
|
self = [super init];
|
|
if (!self) {
|
|
return nil;
|
|
}
|
|
|
|
self.stringEncoding = encoding;
|
|
self.HTTPBodyParts = [NSMutableArray array];
|
|
self.numberOfBytesInPacket = NSIntegerMax;
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void)setInitialAndFinalBoundaries {
|
|
if ([self.HTTPBodyParts count] > 0) {
|
|
for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts) {
|
|
bodyPart.hasInitialBoundary = NO;
|
|
bodyPart.hasFinalBoundary = NO;
|
|
}
|
|
|
|
[[self.HTTPBodyParts objectAtIndex:0] setHasInitialBoundary:YES];
|
|
[[self.HTTPBodyParts lastObject] setHasFinalBoundary:YES];
|
|
}
|
|
}
|
|
|
|
- (void)appendHTTPBodyPart:(AFHTTPBodyPart *)bodyPart {
|
|
[self.HTTPBodyParts addObject:bodyPart];
|
|
}
|
|
|
|
- (BOOL)isEmpty {
|
|
return [self.HTTPBodyParts count] == 0;
|
|
}
|
|
|
|
#pragma mark - NSInputStream
|
|
|
|
- (NSInteger)read:(uint8_t *)buffer maxLength:(NSUInteger)length {
|
|
if ([self streamStatus] == NSStreamStatusClosed) {
|
|
return 0;
|
|
}
|
|
NSInteger bytesRead = 0;
|
|
|
|
while ((NSUInteger)bytesRead < MIN(length, self.numberOfBytesInPacket)) {
|
|
if (!self.currentHTTPBodyPart || ![self.currentHTTPBodyPart hasBytesAvailable]) {
|
|
if (!(self.currentHTTPBodyPart = [self.HTTPBodyPartEnumerator nextObject])) {
|
|
break;
|
|
}
|
|
} else {
|
|
bytesRead += [self.currentHTTPBodyPart read:&buffer[bytesRead] maxLength:(length - (NSUInteger)bytesRead)];
|
|
if (self.delay > 0.0f) {
|
|
[NSThread sleepForTimeInterval:self.delay];
|
|
}
|
|
}
|
|
}
|
|
return bytesRead;
|
|
}
|
|
|
|
- (BOOL)getBuffer:(__unused uint8_t **)buffer length:(__unused NSUInteger *)len {
|
|
return NO;
|
|
}
|
|
|
|
- (BOOL)hasBytesAvailable {
|
|
return [self streamStatus] == NSStreamStatusOpen;
|
|
}
|
|
|
|
#pragma mark - NSStream
|
|
|
|
- (void)open {
|
|
if (self.streamStatus == NSStreamStatusOpen) {
|
|
return;
|
|
}
|
|
|
|
self.streamStatus = NSStreamStatusOpen;
|
|
|
|
[self setInitialAndFinalBoundaries];
|
|
self.HTTPBodyPartEnumerator = [self.HTTPBodyParts objectEnumerator];
|
|
}
|
|
|
|
- (void)close {
|
|
self.streamStatus = NSStreamStatusClosed;
|
|
}
|
|
|
|
- (id)propertyForKey:(__unused NSString *)key {
|
|
return nil;
|
|
}
|
|
|
|
- (BOOL)setProperty:(__unused id)property forKey:(__unused NSString *)key {
|
|
return NO;
|
|
}
|
|
|
|
- (void)scheduleInRunLoop:(__unused NSRunLoop *)aRunLoop
|
|
forMode:(__unused NSString *)mode
|
|
{}
|
|
|
|
- (void)removeFromRunLoop:(__unused NSRunLoop *)aRunLoop
|
|
forMode:(__unused NSString *)mode
|
|
{}
|
|
|
|
- (unsigned long long)contentLength {
|
|
unsigned long long length = 0;
|
|
for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts) {
|
|
length += [bodyPart contentLength];
|
|
}
|
|
|
|
return length;
|
|
}
|
|
|
|
#pragma mark - Undocumented CFReadStream Bridged Methods
|
|
|
|
- (void)_scheduleInCFRunLoop:(__unused CFRunLoopRef)aRunLoop
|
|
forMode:(__unused CFStringRef)aMode
|
|
{}
|
|
|
|
- (void)_unscheduleFromCFRunLoop:(__unused CFRunLoopRef)aRunLoop
|
|
forMode:(__unused CFStringRef)aMode
|
|
{}
|
|
|
|
- (BOOL)_setCFClientFlags:(__unused CFOptionFlags)inFlags
|
|
callback:(__unused CFReadStreamClientCallBack)inCallback
|
|
context:(__unused CFStreamClientContext *)inContext {
|
|
return NO;
|
|
}
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
typedef enum {
|
|
AFEncapsulationBoundaryPhase = 1,
|
|
AFHeaderPhase = 2,
|
|
AFBodyPhase = 3,
|
|
AFFinalBoundaryPhase = 4,
|
|
} AFHTTPBodyPartReadPhase;
|
|
|
|
@interface AFHTTPBodyPart () {
|
|
AFHTTPBodyPartReadPhase _phase;
|
|
unsigned long long _phaseReadOffset;
|
|
}
|
|
|
|
- (BOOL)transitionToNextPhase;
|
|
@end
|
|
|
|
@implementation AFHTTPBodyPart
|
|
@synthesize stringEncoding = _stringEncoding;
|
|
@synthesize headers = _headers;
|
|
@synthesize bodyContentLength = _bodyContentLength;
|
|
@synthesize inputStream = _inputStream;
|
|
@synthesize hasInitialBoundary = _hasInitialBoundary;
|
|
@synthesize hasFinalBoundary = _hasFinalBoundary;
|
|
|
|
- (id)init {
|
|
self = [super init];
|
|
if (!self) {
|
|
return nil;
|
|
}
|
|
|
|
[self transitionToNextPhase];
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc {
|
|
if (_inputStream) {
|
|
[_inputStream close];
|
|
_inputStream = nil;
|
|
}
|
|
}
|
|
|
|
- (NSString *)stringForHeaders {
|
|
NSMutableString *headerString = [NSMutableString string];
|
|
for (NSString *field in [self.headers allKeys]) {
|
|
[headerString appendString:[NSString stringWithFormat:@"%@: %@%@", field, [self.headers valueForKey:field], kAFMultipartFormCRLF]];
|
|
}
|
|
[headerString appendString:kAFMultipartFormCRLF];
|
|
|
|
return [NSString stringWithString:headerString];
|
|
}
|
|
|
|
- (unsigned long long)contentLength {
|
|
unsigned long long length = 0;
|
|
|
|
NSData *encapsulationBoundaryData = [([self hasInitialBoundary] ? AFMultipartFormInitialBoundary() : AFMultipartFormEncapsulationBoundary()) dataUsingEncoding:self.stringEncoding];
|
|
length += [encapsulationBoundaryData length];
|
|
|
|
NSData *headersData = [[self stringForHeaders] dataUsingEncoding:self.stringEncoding];
|
|
length += [headersData length];
|
|
|
|
length += _bodyContentLength;
|
|
|
|
NSData *closingBoundaryData = ([self hasFinalBoundary] ? [AFMultipartFormFinalBoundary() dataUsingEncoding:self.stringEncoding] : [NSData data]);
|
|
length += [closingBoundaryData length];
|
|
|
|
return length;
|
|
}
|
|
|
|
- (BOOL)hasBytesAvailable {
|
|
// Allows `read:maxLength:` to be called again if `AFMultipartFormFinalBoundary` doesn't fit into the avaiable buffer
|
|
if (_phase == AFFinalBoundaryPhase) {
|
|
return YES;
|
|
}
|
|
|
|
switch (self.inputStream.streamStatus) {
|
|
case NSStreamStatusNotOpen:
|
|
case NSStreamStatusOpening:
|
|
case NSStreamStatusOpen:
|
|
case NSStreamStatusReading:
|
|
case NSStreamStatusWriting:
|
|
return YES;
|
|
case NSStreamStatusAtEnd:
|
|
case NSStreamStatusClosed:
|
|
case NSStreamStatusError:
|
|
return NO;
|
|
}
|
|
}
|
|
|
|
- (NSInteger)read:(uint8_t *)buffer maxLength:(NSUInteger)length {
|
|
NSInteger bytesRead = 0;
|
|
|
|
if (_phase == AFEncapsulationBoundaryPhase) {
|
|
NSData *encapsulationBoundaryData = [([self hasInitialBoundary] ? AFMultipartFormInitialBoundary() : AFMultipartFormEncapsulationBoundary()) dataUsingEncoding:self.stringEncoding];
|
|
bytesRead += [self readData:encapsulationBoundaryData intoBuffer:&buffer[bytesRead] maxLength:(length - (NSUInteger)bytesRead)];
|
|
}
|
|
|
|
if (_phase == AFHeaderPhase) {
|
|
NSData *headersData = [[self stringForHeaders] dataUsingEncoding:self.stringEncoding];
|
|
bytesRead += [self readData:headersData intoBuffer:&buffer[bytesRead] maxLength:(length - (NSUInteger)bytesRead)];
|
|
}
|
|
|
|
if (_phase == AFBodyPhase) {
|
|
if ([self.inputStream hasBytesAvailable]) {
|
|
bytesRead += [self.inputStream read:&buffer[bytesRead] maxLength:(length - (NSUInteger)bytesRead)];
|
|
}
|
|
|
|
if (![self.inputStream hasBytesAvailable]) {
|
|
[self transitionToNextPhase];
|
|
}
|
|
}
|
|
|
|
if (_phase == AFFinalBoundaryPhase) {
|
|
NSData *closingBoundaryData = ([self hasFinalBoundary] ? [AFMultipartFormFinalBoundary() dataUsingEncoding:self.stringEncoding] : [NSData data]);
|
|
bytesRead += [self readData:closingBoundaryData intoBuffer:&buffer[bytesRead] maxLength:(length - (NSUInteger)bytesRead)];
|
|
}
|
|
|
|
return bytesRead;
|
|
}
|
|
|
|
- (NSInteger)readData:(NSData *)data
|
|
intoBuffer:(uint8_t *)buffer
|
|
maxLength:(NSUInteger)length
|
|
{
|
|
NSRange range = NSMakeRange((NSUInteger)_phaseReadOffset, MIN([data length] - ((NSUInteger)_phaseReadOffset), length));
|
|
[data getBytes:buffer range:range];
|
|
|
|
_phaseReadOffset += range.length;
|
|
|
|
if (((NSUInteger)_phaseReadOffset) >= [data length]) {
|
|
[self transitionToNextPhase];
|
|
}
|
|
|
|
return (NSInteger)range.length;
|
|
}
|
|
|
|
- (BOOL)transitionToNextPhase {
|
|
if (![[NSThread currentThread] isMainThread]) {
|
|
[self performSelectorOnMainThread:@selector(transitionToNextPhase) withObject:nil waitUntilDone:YES];
|
|
return YES;
|
|
}
|
|
|
|
switch (_phase) {
|
|
case AFEncapsulationBoundaryPhase:
|
|
_phase = AFHeaderPhase;
|
|
break;
|
|
case AFHeaderPhase:
|
|
[self.inputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSRunLoopCommonModes];
|
|
[self.inputStream open];
|
|
_phase = AFBodyPhase;
|
|
break;
|
|
case AFBodyPhase:
|
|
[self.inputStream close];
|
|
_phase = AFFinalBoundaryPhase;
|
|
break;
|
|
case AFFinalBoundaryPhase:
|
|
default:
|
|
_phase = AFEncapsulationBoundaryPhase;
|
|
break;
|
|
}
|
|
_phaseReadOffset = 0;
|
|
|
|
return YES;
|
|
}
|
|
|
|
@end
|