PageRenderTime 49ms CodeModel.GetById 12ms RepoModel.GetById 0ms app.codeStats 0ms

/TeamTalk/Pods/AFNetworking/AFNetworking/AFURLRequestSerialization.m

https://gitlab.com/lisit1003/TTiOSClient
Objective C | 1281 lines | 973 code | 281 blank | 27 comment | 120 complexity | 78fd9a9c30b11c351b2c27cc3f00be54 MD5 | raw file
  1. // AFSerialization.h
  2. //
  3. // Copyright (c) 2013-2014 AFNetworking (http://afnetworking.com)
  4. //
  5. // Permission is hereby granted, free of charge, to any person obtaining a copy
  6. // of this software and associated documentation files (the "Software"), to deal
  7. // in the Software without restriction, including without limitation the rights
  8. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. // copies of the Software, and to permit persons to whom the Software is
  10. // furnished to do so, subject to the following conditions:
  11. //
  12. // The above copyright notice and this permission notice shall be included in
  13. // all copies or substantial portions of the Software.
  14. //
  15. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21. // THE SOFTWARE.
  22. #import "AFURLRequestSerialization.h"
  23. #if __IPHONE_OS_VERSION_MIN_REQUIRED
  24. #import <MobileCoreServices/MobileCoreServices.h>
  25. #else
  26. #import <CoreServices/CoreServices.h>
  27. #endif
  28. NSString * const AFURLRequestSerializationErrorDomain = @"com.alamofire.error.serialization.request";
  29. NSString * const AFNetworkingOperationFailingURLRequestErrorKey = @"com.alamofire.serialization.request.error.response";
  30. typedef NSString * (^AFQueryStringSerializationBlock)(NSURLRequest *request, NSDictionary *parameters, NSError *__autoreleasing *error);
  31. static NSString * AFBase64EncodedStringFromString(NSString *string) {
  32. NSData *data = [NSData dataWithBytes:[string UTF8String] length:[string lengthOfBytesUsingEncoding:NSUTF8StringEncoding]];
  33. NSUInteger length = [data length];
  34. NSMutableData *mutableData = [NSMutableData dataWithLength:((length + 2) / 3) * 4];
  35. uint8_t *input = (uint8_t *)[data bytes];
  36. uint8_t *output = (uint8_t *)[mutableData mutableBytes];
  37. for (NSUInteger i = 0; i < length; i += 3) {
  38. NSUInteger value = 0;
  39. for (NSUInteger j = i; j < (i + 3); j++) {
  40. value <<= 8;
  41. if (j < length) {
  42. value |= (0xFF & input[j]);
  43. }
  44. }
  45. static uint8_t const kAFBase64EncodingTable[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
  46. NSUInteger idx = (i / 3) * 4;
  47. output[idx + 0] = kAFBase64EncodingTable[(value >> 18) & 0x3F];
  48. output[idx + 1] = kAFBase64EncodingTable[(value >> 12) & 0x3F];
  49. output[idx + 2] = (i + 1) < length ? kAFBase64EncodingTable[(value >> 6) & 0x3F] : '=';
  50. output[idx + 3] = (i + 2) < length ? kAFBase64EncodingTable[(value >> 0) & 0x3F] : '=';
  51. }
  52. return [[NSString alloc] initWithData:mutableData encoding:NSASCIIStringEncoding];
  53. }
  54. static NSString * const kAFCharactersToBeEscapedInQueryString = @":/?&=;+!@#$()',*";
  55. static NSString * AFPercentEscapedQueryStringKeyFromStringWithEncoding(NSString *string, NSStringEncoding encoding) {
  56. static NSString * const kAFCharactersToLeaveUnescapedInQueryStringPairKey = @"[].";
  57. return (__bridge_transfer NSString *)CFURLCreateStringByAddingPercentEscapes(kCFAllocatorDefault, (__bridge CFStringRef)string, (__bridge CFStringRef)kAFCharactersToLeaveUnescapedInQueryStringPairKey, (__bridge CFStringRef)kAFCharactersToBeEscapedInQueryString, CFStringConvertNSStringEncodingToEncoding(encoding));
  58. }
  59. static NSString * AFPercentEscapedQueryStringValueFromStringWithEncoding(NSString *string, NSStringEncoding encoding) {
  60. return (__bridge_transfer NSString *)CFURLCreateStringByAddingPercentEscapes(kCFAllocatorDefault, (__bridge CFStringRef)string, NULL, (__bridge CFStringRef)kAFCharactersToBeEscapedInQueryString, CFStringConvertNSStringEncodingToEncoding(encoding));
  61. }
  62. #pragma mark -
  63. @interface AFQueryStringPair : NSObject
  64. @property (readwrite, nonatomic, strong) id field;
  65. @property (readwrite, nonatomic, strong) id value;
  66. - (id)initWithField:(id)field value:(id)value;
  67. - (NSString *)URLEncodedStringValueWithEncoding:(NSStringEncoding)stringEncoding;
  68. @end
  69. @implementation AFQueryStringPair
  70. - (id)initWithField:(id)field value:(id)value {
  71. self = [super init];
  72. if (!self) {
  73. return nil;
  74. }
  75. self.field = field;
  76. self.value = value;
  77. return self;
  78. }
  79. - (NSString *)URLEncodedStringValueWithEncoding:(NSStringEncoding)stringEncoding {
  80. if (!self.value || [self.value isEqual:[NSNull null]]) {
  81. return AFPercentEscapedQueryStringKeyFromStringWithEncoding([self.field description], stringEncoding);
  82. } else {
  83. return [NSString stringWithFormat:@"%@=%@", AFPercentEscapedQueryStringKeyFromStringWithEncoding([self.field description], stringEncoding), AFPercentEscapedQueryStringValueFromStringWithEncoding([self.value description], stringEncoding)];
  84. }
  85. }
  86. @end
  87. #pragma mark -
  88. extern NSArray * AFQueryStringPairsFromDictionary(NSDictionary *dictionary);
  89. extern NSArray * AFQueryStringPairsFromKeyAndValue(NSString *key, id value);
  90. static NSString * AFQueryStringFromParametersWithEncoding(NSDictionary *parameters, NSStringEncoding stringEncoding) {
  91. NSMutableArray *mutablePairs = [NSMutableArray array];
  92. for (AFQueryStringPair *pair in AFQueryStringPairsFromDictionary(parameters)) {
  93. [mutablePairs addObject:[pair URLEncodedStringValueWithEncoding:stringEncoding]];
  94. }
  95. return [mutablePairs componentsJoinedByString:@"&"];
  96. }
  97. NSArray * AFQueryStringPairsFromDictionary(NSDictionary *dictionary) {
  98. return AFQueryStringPairsFromKeyAndValue(nil, dictionary);
  99. }
  100. NSArray * AFQueryStringPairsFromKeyAndValue(NSString *key, id value) {
  101. NSMutableArray *mutableQueryStringComponents = [NSMutableArray array];
  102. NSSortDescriptor *sortDescriptor = [NSSortDescriptor sortDescriptorWithKey:@"description" ascending:YES selector:@selector(compare:)];
  103. if ([value isKindOfClass:[NSDictionary class]]) {
  104. NSDictionary *dictionary = value;
  105. // Sort dictionary keys to ensure consistent ordering in query string, which is important when deserializing potentially ambiguous sequences, such as an array of dictionaries
  106. for (id nestedKey in [dictionary.allKeys sortedArrayUsingDescriptors:@[ sortDescriptor ]]) {
  107. id nestedValue = [dictionary objectForKey:nestedKey];
  108. if (nestedValue) {
  109. [mutableQueryStringComponents addObjectsFromArray:AFQueryStringPairsFromKeyAndValue((key ? [NSString stringWithFormat:@"%@[%@]", key, nestedKey] : nestedKey), nestedValue)];
  110. }
  111. }
  112. } else if ([value isKindOfClass:[NSArray class]]) {
  113. NSArray *array = value;
  114. for (id nestedValue in array) {
  115. [mutableQueryStringComponents addObjectsFromArray:AFQueryStringPairsFromKeyAndValue([NSString stringWithFormat:@"%@[]", key], nestedValue)];
  116. }
  117. } else if ([value isKindOfClass:[NSSet class]]) {
  118. NSSet *set = value;
  119. for (id obj in [set sortedArrayUsingDescriptors:@[ sortDescriptor ]]) {
  120. [mutableQueryStringComponents addObjectsFromArray:AFQueryStringPairsFromKeyAndValue(key, obj)];
  121. }
  122. } else {
  123. [mutableQueryStringComponents addObject:[[AFQueryStringPair alloc] initWithField:key value:value]];
  124. }
  125. return mutableQueryStringComponents;
  126. }
  127. #pragma mark -
  128. @interface AFStreamingMultipartFormData : NSObject <AFMultipartFormData>
  129. - (instancetype)initWithURLRequest:(NSMutableURLRequest *)urlRequest
  130. stringEncoding:(NSStringEncoding)encoding;
  131. - (NSMutableURLRequest *)requestByFinalizingMultipartFormData;
  132. @end
  133. #pragma mark -
  134. @interface AFHTTPRequestSerializer ()
  135. @property (readwrite, nonatomic, strong) NSMutableDictionary *mutableHTTPRequestHeaders;
  136. @property (readwrite, nonatomic, assign) AFHTTPRequestQueryStringSerializationStyle queryStringSerializationStyle;
  137. @property (readwrite, nonatomic, copy) AFQueryStringSerializationBlock queryStringSerialization;
  138. @end
  139. @implementation AFHTTPRequestSerializer
  140. + (instancetype)serializer {
  141. return [[self alloc] init];
  142. }
  143. - (instancetype)init {
  144. self = [super init];
  145. if (!self) {
  146. return nil;
  147. }
  148. self.stringEncoding = NSUTF8StringEncoding;
  149. self.allowsCellularAccess = YES;
  150. self.cachePolicy = NSURLRequestUseProtocolCachePolicy;
  151. self.HTTPShouldHandleCookies = YES;
  152. self.HTTPShouldUsePipelining = NO;
  153. self.networkServiceType = NSURLNetworkServiceTypeDefault;
  154. self.timeoutInterval = 60;
  155. self.mutableHTTPRequestHeaders = [NSMutableDictionary dictionary];
  156. // Accept-Language HTTP Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.4
  157. NSMutableArray *acceptLanguagesComponents = [NSMutableArray array];
  158. [[NSLocale preferredLanguages] enumerateObjectsUsingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
  159. float q = 1.0f - (idx * 0.1f);
  160. [acceptLanguagesComponents addObject:[NSString stringWithFormat:@"%@;q=%0.1g", obj, q]];
  161. *stop = q <= 0.5f;
  162. }];
  163. [self setValue:[acceptLanguagesComponents componentsJoinedByString:@", "] forHTTPHeaderField:@"Accept-Language"];
  164. NSString *userAgent = nil;
  165. #pragma clang diagnostic push
  166. #pragma clang diagnostic ignored "-Wgnu"
  167. #if defined(__IPHONE_OS_VERSION_MIN_REQUIRED)
  168. // User-Agent Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.43
  169. userAgent = [NSString stringWithFormat:@"%@/%@ (%@; iOS %@; Scale/%0.2f)", [[[NSBundle mainBundle] infoDictionary] objectForKey:(__bridge NSString *)kCFBundleExecutableKey] ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(__bridge NSString *)kCFBundleIdentifierKey], (__bridge id)CFBundleGetValueForInfoDictionaryKey(CFBundleGetMainBundle(), kCFBundleVersionKey) ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(__bridge NSString *)kCFBundleVersionKey], [[UIDevice currentDevice] model], [[UIDevice currentDevice] systemVersion], [[UIScreen mainScreen] scale]];
  170. #elif defined(__MAC_OS_X_VERSION_MIN_REQUIRED)
  171. userAgent = [NSString stringWithFormat:@"%@/%@ (Mac OS X %@)", [[[NSBundle mainBundle] infoDictionary] objectForKey:(__bridge NSString *)kCFBundleExecutableKey] ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(__bridge NSString *)kCFBundleIdentifierKey], [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleShortVersionString"] ?: [[[NSBundle mainBundle] infoDictionary] objectForKey:(__bridge NSString *)kCFBundleVersionKey], [[NSProcessInfo processInfo] operatingSystemVersionString]];
  172. #endif
  173. #pragma clang diagnostic pop
  174. if (userAgent) {
  175. if (![userAgent canBeConvertedToEncoding:NSASCIIStringEncoding]) {
  176. NSMutableString *mutableUserAgent = [userAgent mutableCopy];
  177. if (CFStringTransform((__bridge CFMutableStringRef)(mutableUserAgent), NULL, (__bridge CFStringRef)@"Any-Latin; Latin-ASCII; [:^ASCII:] Remove", false)) {
  178. userAgent = mutableUserAgent;
  179. }
  180. }
  181. [self setValue:userAgent forHTTPHeaderField:@"User-Agent"];
  182. }
  183. // HTTP Method Definitions; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec9.html
  184. self.HTTPMethodsEncodingParametersInURI = [NSSet setWithObjects:@"GET", @"HEAD", @"DELETE", nil];
  185. return self;
  186. }
  187. #pragma mark -
  188. - (NSDictionary *)HTTPRequestHeaders {
  189. return [NSDictionary dictionaryWithDictionary:self.mutableHTTPRequestHeaders];
  190. }
  191. - (void)setValue:(NSString *)value forHTTPHeaderField:(NSString *)field {
  192. [self.mutableHTTPRequestHeaders setValue:value forKey:field];
  193. }
  194. - (void)setAuthorizationHeaderFieldWithUsername:(NSString *)username password:(NSString *)password {
  195. NSString *basicAuthCredentials = [NSString stringWithFormat:@"%@:%@", username, password];
  196. [self setValue:[NSString stringWithFormat:@"Basic %@", AFBase64EncodedStringFromString(basicAuthCredentials)] forHTTPHeaderField:@"Authorization"];
  197. }
  198. - (void)setAuthorizationHeaderFieldWithToken:(NSString *)token {
  199. [self setValue:[NSString stringWithFormat:@"Token token=\"%@\"", token] forHTTPHeaderField:@"Authorization"];
  200. }
  201. - (void)clearAuthorizationHeader {
  202. [self.mutableHTTPRequestHeaders removeObjectForKey:@"Authorization"];
  203. }
  204. #pragma mark -
  205. - (void)setQueryStringSerializationWithStyle:(AFHTTPRequestQueryStringSerializationStyle)style {
  206. self.queryStringSerializationStyle = style;
  207. self.queryStringSerialization = nil;
  208. }
  209. - (void)setQueryStringSerializationWithBlock:(NSString *(^)(NSURLRequest *, NSDictionary *, NSError *__autoreleasing *))block {
  210. self.queryStringSerialization = block;
  211. }
  212. #pragma mark -
  213. - (NSMutableURLRequest *)requestWithMethod:(NSString *)method
  214. URLString:(NSString *)URLString
  215. parameters:(id)parameters
  216. {
  217. return [self requestWithMethod:method URLString:URLString parameters:parameters error:nil];
  218. }
  219. - (NSMutableURLRequest *)requestWithMethod:(NSString *)method
  220. URLString:(NSString *)URLString
  221. parameters:(id)parameters
  222. error:(NSError *__autoreleasing *)error
  223. {
  224. NSParameterAssert(method);
  225. NSParameterAssert(URLString);
  226. NSURL *url = [NSURL URLWithString:URLString];
  227. NSParameterAssert(url);
  228. NSMutableURLRequest *mutableRequest = [[NSMutableURLRequest alloc] initWithURL:url];
  229. mutableRequest.HTTPMethod = method;
  230. mutableRequest.allowsCellularAccess = self.allowsCellularAccess;
  231. mutableRequest.cachePolicy = self.cachePolicy;
  232. mutableRequest.HTTPShouldHandleCookies = self.HTTPShouldHandleCookies;
  233. mutableRequest.HTTPShouldUsePipelining = self.HTTPShouldUsePipelining;
  234. mutableRequest.networkServiceType = self.networkServiceType;
  235. mutableRequest.timeoutInterval = self.timeoutInterval;
  236. mutableRequest = [[self requestBySerializingRequest:mutableRequest withParameters:parameters error:error] mutableCopy];
  237. return mutableRequest;
  238. }
  239. - (NSMutableURLRequest *)multipartFormRequestWithMethod:(NSString *)method
  240. URLString:(NSString *)URLString
  241. parameters:(NSDictionary *)parameters
  242. constructingBodyWithBlock:(void (^)(id <AFMultipartFormData> formData))block
  243. {
  244. return [self multipartFormRequestWithMethod:method URLString:URLString parameters:parameters constructingBodyWithBlock:block error:nil];
  245. }
  246. - (NSMutableURLRequest *)multipartFormRequestWithMethod:(NSString *)method
  247. URLString:(NSString *)URLString
  248. parameters:(NSDictionary *)parameters
  249. constructingBodyWithBlock:(void (^)(id <AFMultipartFormData> formData))block
  250. error:(NSError *__autoreleasing *)error
  251. {
  252. NSParameterAssert(method);
  253. NSParameterAssert(![method isEqualToString:@"GET"] && ![method isEqualToString:@"HEAD"]);
  254. NSMutableURLRequest *mutableRequest = [self requestWithMethod:method URLString:URLString parameters:nil error:error];
  255. __block AFStreamingMultipartFormData *formData = [[AFStreamingMultipartFormData alloc] initWithURLRequest:mutableRequest stringEncoding:NSUTF8StringEncoding];
  256. if (parameters) {
  257. for (AFQueryStringPair *pair in AFQueryStringPairsFromDictionary(parameters)) {
  258. NSData *data = nil;
  259. if ([pair.value isKindOfClass:[NSData class]]) {
  260. data = pair.value;
  261. } else if ([pair.value isEqual:[NSNull null]]) {
  262. data = [NSData data];
  263. } else {
  264. data = [[pair.value description] dataUsingEncoding:self.stringEncoding];
  265. }
  266. if (data) {
  267. [formData appendPartWithFormData:data name:[pair.field description]];
  268. }
  269. }
  270. }
  271. if (block) {
  272. block(formData);
  273. }
  274. return [formData requestByFinalizingMultipartFormData];
  275. }
  276. - (NSMutableURLRequest *)requestWithMultipartFormRequest:(NSURLRequest *)request
  277. writingStreamContentsToFile:(NSURL *)fileURL
  278. completionHandler:(void (^)(NSError *error))handler
  279. {
  280. if (!request.HTTPBodyStream) {
  281. return [request mutableCopy];
  282. }
  283. NSParameterAssert([fileURL isFileURL]);
  284. NSInputStream *inputStream = request.HTTPBodyStream;
  285. NSOutputStream *outputStream = [[NSOutputStream alloc] initWithURL:fileURL append:NO];
  286. __block NSError *error = nil;
  287. dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
  288. [inputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
  289. [outputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
  290. [inputStream open];
  291. [outputStream open];
  292. while ([inputStream hasBytesAvailable] && [outputStream hasSpaceAvailable]) {
  293. uint8_t buffer[1024];
  294. NSInteger bytesRead = [inputStream read:buffer maxLength:1024];
  295. if (inputStream.streamError || bytesRead < 0) {
  296. error = inputStream.streamError;
  297. break;
  298. }
  299. NSInteger bytesWritten = [outputStream write:buffer maxLength:(NSUInteger)bytesRead];
  300. if (outputStream.streamError || bytesWritten < 0) {
  301. error = outputStream.streamError;
  302. break;
  303. }
  304. if (bytesRead == 0 && bytesWritten == 0) {
  305. break;
  306. }
  307. }
  308. [outputStream close];
  309. [inputStream close];
  310. if (handler) {
  311. dispatch_async(dispatch_get_main_queue(), ^{
  312. handler(error);
  313. });
  314. }
  315. });
  316. NSMutableURLRequest *mutableRequest = [request mutableCopy];
  317. mutableRequest.HTTPBodyStream = nil;
  318. return mutableRequest;
  319. }
  320. #pragma mark - AFURLRequestSerialization
  321. - (NSURLRequest *)requestBySerializingRequest:(NSURLRequest *)request
  322. withParameters:(id)parameters
  323. error:(NSError *__autoreleasing *)error
  324. {
  325. NSParameterAssert(request);
  326. NSMutableURLRequest *mutableRequest = [request mutableCopy];
  327. [self.HTTPRequestHeaders enumerateKeysAndObjectsUsingBlock:^(id field, id value, BOOL * __unused stop) {
  328. if (![request valueForHTTPHeaderField:field]) {
  329. [mutableRequest setValue:value forHTTPHeaderField:field];
  330. }
  331. }];
  332. if (parameters) {
  333. NSString *query = nil;
  334. if (self.queryStringSerialization) {
  335. query = self.queryStringSerialization(request, parameters, error);
  336. } else {
  337. switch (self.queryStringSerializationStyle) {
  338. case AFHTTPRequestQueryStringDefaultStyle:
  339. query = AFQueryStringFromParametersWithEncoding(parameters, self.stringEncoding);
  340. break;
  341. }
  342. }
  343. if ([self.HTTPMethodsEncodingParametersInURI containsObject:[[request HTTPMethod] uppercaseString]]) {
  344. mutableRequest.URL = [NSURL URLWithString:[[mutableRequest.URL absoluteString] stringByAppendingFormat:mutableRequest.URL.query ? @"&%@" : @"?%@", query]];
  345. } else {
  346. if (![mutableRequest valueForHTTPHeaderField:@"Content-Type"]) {
  347. NSString *charset = (__bridge NSString *)CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(self.stringEncoding));
  348. [mutableRequest setValue:[NSString stringWithFormat:@"application/x-www-form-urlencoded; charset=%@", charset] forHTTPHeaderField:@"Content-Type"];
  349. }
  350. [mutableRequest setHTTPBody:[query dataUsingEncoding:self.stringEncoding]];
  351. }
  352. }
  353. return mutableRequest;
  354. }
  355. #pragma mark - NSecureCoding
  356. + (BOOL)supportsSecureCoding {
  357. return YES;
  358. }
  359. - (id)initWithCoder:(NSCoder *)decoder {
  360. self = [self init];
  361. if (!self) {
  362. return nil;
  363. }
  364. self.mutableHTTPRequestHeaders = [[decoder decodeObjectOfClass:[NSDictionary class] forKey:NSStringFromSelector(@selector(mutableHTTPRequestHeaders))] mutableCopy];
  365. self.queryStringSerializationStyle = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(queryStringSerializationStyle))] unsignedIntegerValue];
  366. return self;
  367. }
  368. - (void)encodeWithCoder:(NSCoder *)coder {
  369. [coder encodeObject:self.mutableHTTPRequestHeaders forKey:NSStringFromSelector(@selector(mutableHTTPRequestHeaders))];
  370. [coder encodeInteger:self.queryStringSerializationStyle forKey:NSStringFromSelector(@selector(queryStringSerializationStyle))];
  371. }
  372. #pragma mark - NSCopying
  373. - (id)copyWithZone:(NSZone *)zone {
  374. AFHTTPRequestSerializer *serializer = [[[self class] allocWithZone:zone] init];
  375. serializer.mutableHTTPRequestHeaders = [self.mutableHTTPRequestHeaders mutableCopyWithZone:zone];
  376. serializer.queryStringSerializationStyle = self.queryStringSerializationStyle;
  377. serializer.queryStringSerialization = self.queryStringSerialization;
  378. return serializer;
  379. }
  380. @end
  381. #pragma mark -
  382. static NSString * AFCreateMultipartFormBoundary() {
  383. return [NSString stringWithFormat:@"Boundary+%08X%08X", arc4random(), arc4random()];
  384. }
  385. static NSString * const kAFMultipartFormCRLF = @"\r\n";
  386. static inline NSString * AFMultipartFormInitialBoundary(NSString *boundary) {
  387. return [NSString stringWithFormat:@"--%@%@", boundary, kAFMultipartFormCRLF];
  388. }
  389. static inline NSString * AFMultipartFormEncapsulationBoundary(NSString *boundary) {
  390. return [NSString stringWithFormat:@"%@--%@%@", kAFMultipartFormCRLF, boundary, kAFMultipartFormCRLF];
  391. }
  392. static inline NSString * AFMultipartFormFinalBoundary(NSString *boundary) {
  393. return [NSString stringWithFormat:@"%@--%@--%@", kAFMultipartFormCRLF, boundary, kAFMultipartFormCRLF];
  394. }
  395. static inline NSString * AFContentTypeForPathExtension(NSString *extension) {
  396. #ifdef __UTTYPE__
  397. NSString *UTI = (__bridge_transfer NSString *)UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, (__bridge CFStringRef)extension, NULL);
  398. NSString *contentType = (__bridge_transfer NSString *)UTTypeCopyPreferredTagWithClass((__bridge CFStringRef)UTI, kUTTagClassMIMEType);
  399. if (!contentType) {
  400. return @"application/octet-stream";
  401. } else {
  402. return contentType;
  403. }
  404. #else
  405. #pragma unused (extension)
  406. return @"application/octet-stream";
  407. #endif
  408. }
  409. NSUInteger const kAFUploadStream3GSuggestedPacketSize = 1024 * 16;
  410. NSTimeInterval const kAFUploadStream3GSuggestedDelay = 0.2;
  411. @interface AFHTTPBodyPart : NSObject
  412. @property (nonatomic, assign) NSStringEncoding stringEncoding;
  413. @property (nonatomic, strong) NSDictionary *headers;
  414. @property (nonatomic, copy) NSString *boundary;
  415. @property (nonatomic, strong) id body;
  416. @property (nonatomic, assign) unsigned long long bodyContentLength;
  417. @property (nonatomic, strong) NSInputStream *inputStream;
  418. @property (nonatomic, assign) BOOL hasInitialBoundary;
  419. @property (nonatomic, assign) BOOL hasFinalBoundary;
  420. @property (readonly, nonatomic, assign, getter = hasBytesAvailable) BOOL bytesAvailable;
  421. @property (readonly, nonatomic, assign) unsigned long long contentLength;
  422. - (NSInteger)read:(uint8_t *)buffer
  423. maxLength:(NSUInteger)length;
  424. @end
  425. @interface AFMultipartBodyStream : NSInputStream <NSStreamDelegate>
  426. @property (nonatomic, assign) NSUInteger numberOfBytesInPacket;
  427. @property (nonatomic, assign) NSTimeInterval delay;
  428. @property (nonatomic, strong) NSInputStream *inputStream;
  429. @property (readonly, nonatomic, assign) unsigned long long contentLength;
  430. @property (readonly, nonatomic, assign, getter = isEmpty) BOOL empty;
  431. - (id)initWithStringEncoding:(NSStringEncoding)encoding;
  432. - (void)setInitialAndFinalBoundaries;
  433. - (void)appendHTTPBodyPart:(AFHTTPBodyPart *)bodyPart;
  434. @end
  435. #pragma mark -
  436. @interface AFStreamingMultipartFormData ()
  437. @property (readwrite, nonatomic, copy) NSMutableURLRequest *request;
  438. @property (readwrite, nonatomic, assign) NSStringEncoding stringEncoding;
  439. @property (readwrite, nonatomic, copy) NSString *boundary;
  440. @property (readwrite, nonatomic, strong) AFMultipartBodyStream *bodyStream;
  441. @end
  442. @implementation AFStreamingMultipartFormData
  443. - (id)initWithURLRequest:(NSMutableURLRequest *)urlRequest
  444. stringEncoding:(NSStringEncoding)encoding
  445. {
  446. self = [super init];
  447. if (!self) {
  448. return nil;
  449. }
  450. self.request = urlRequest;
  451. self.stringEncoding = encoding;
  452. self.boundary = AFCreateMultipartFormBoundary();
  453. self.bodyStream = [[AFMultipartBodyStream alloc] initWithStringEncoding:encoding];
  454. return self;
  455. }
  456. - (BOOL)appendPartWithFileURL:(NSURL *)fileURL
  457. name:(NSString *)name
  458. error:(NSError * __autoreleasing *)error
  459. {
  460. NSParameterAssert(fileURL);
  461. NSParameterAssert(name);
  462. NSString *fileName = [fileURL lastPathComponent];
  463. NSString *mimeType = AFContentTypeForPathExtension([fileURL pathExtension]);
  464. return [self appendPartWithFileURL:fileURL name:name fileName:fileName mimeType:mimeType error:error];
  465. }
  466. - (BOOL)appendPartWithFileURL:(NSURL *)fileURL
  467. name:(NSString *)name
  468. fileName:(NSString *)fileName
  469. mimeType:(NSString *)mimeType
  470. error:(NSError * __autoreleasing *)error
  471. {
  472. NSParameterAssert(fileURL);
  473. NSParameterAssert(name);
  474. NSParameterAssert(fileName);
  475. NSParameterAssert(mimeType);
  476. if (![fileURL isFileURL]) {
  477. NSDictionary *userInfo = @{NSLocalizedFailureReasonErrorKey: NSLocalizedStringFromTable(@"Expected URL to be a file URL", @"AFNetworking", nil)};
  478. if (error) {
  479. *error = [[NSError alloc] initWithDomain:AFURLRequestSerializationErrorDomain code:NSURLErrorBadURL userInfo:userInfo];
  480. }
  481. return NO;
  482. } else if ([fileURL checkResourceIsReachableAndReturnError:error] == NO) {
  483. NSDictionary *userInfo = @{NSLocalizedFailureReasonErrorKey: NSLocalizedStringFromTable(@"File URL not reachable.", @"AFNetworking", nil)};
  484. if (error) {
  485. *error = [[NSError alloc] initWithDomain:AFURLRequestSerializationErrorDomain code:NSURLErrorBadURL userInfo:userInfo];
  486. }
  487. return NO;
  488. }
  489. NSDictionary *fileAttributes = [[NSFileManager defaultManager] attributesOfItemAtPath:[fileURL path] error:error];
  490. if (!fileAttributes) {
  491. return NO;
  492. }
  493. NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
  494. [mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"; filename=\"%@\"", name, fileName] forKey:@"Content-Disposition"];
  495. [mutableHeaders setValue:mimeType forKey:@"Content-Type"];
  496. AFHTTPBodyPart *bodyPart = [[AFHTTPBodyPart alloc] init];
  497. bodyPart.stringEncoding = self.stringEncoding;
  498. bodyPart.headers = mutableHeaders;
  499. bodyPart.boundary = self.boundary;
  500. bodyPart.body = fileURL;
  501. bodyPart.bodyContentLength = [[fileAttributes objectForKey:NSFileSize] unsignedLongLongValue];
  502. [self.bodyStream appendHTTPBodyPart:bodyPart];
  503. return YES;
  504. }
  505. - (void)appendPartWithInputStream:(NSInputStream *)inputStream
  506. name:(NSString *)name
  507. fileName:(NSString *)fileName
  508. length:(int64_t)length
  509. mimeType:(NSString *)mimeType
  510. {
  511. NSParameterAssert(name);
  512. NSParameterAssert(fileName);
  513. NSParameterAssert(mimeType);
  514. NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
  515. [mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"; filename=\"%@\"", name, fileName] forKey:@"Content-Disposition"];
  516. [mutableHeaders setValue:mimeType forKey:@"Content-Type"];
  517. AFHTTPBodyPart *bodyPart = [[AFHTTPBodyPart alloc] init];
  518. bodyPart.stringEncoding = self.stringEncoding;
  519. bodyPart.headers = mutableHeaders;
  520. bodyPart.boundary = self.boundary;
  521. bodyPart.body = inputStream;
  522. bodyPart.bodyContentLength = (unsigned long long)length;
  523. [self.bodyStream appendHTTPBodyPart:bodyPart];
  524. }
  525. - (void)appendPartWithFileData:(NSData *)data
  526. name:(NSString *)name
  527. fileName:(NSString *)fileName
  528. mimeType:(NSString *)mimeType
  529. {
  530. NSParameterAssert(name);
  531. NSParameterAssert(fileName);
  532. NSParameterAssert(mimeType);
  533. NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
  534. [mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"; filename=\"%@\"", name, fileName] forKey:@"Content-Disposition"];
  535. [mutableHeaders setValue:mimeType forKey:@"Content-Type"];
  536. [self appendPartWithHeaders:mutableHeaders body:data];
  537. }
  538. - (void)appendPartWithFormData:(NSData *)data
  539. name:(NSString *)name
  540. {
  541. NSParameterAssert(name);
  542. NSMutableDictionary *mutableHeaders = [NSMutableDictionary dictionary];
  543. [mutableHeaders setValue:[NSString stringWithFormat:@"form-data; name=\"%@\"", name] forKey:@"Content-Disposition"];
  544. [self appendPartWithHeaders:mutableHeaders body:data];
  545. }
  546. - (void)appendPartWithHeaders:(NSDictionary *)headers
  547. body:(NSData *)body
  548. {
  549. NSParameterAssert(body);
  550. AFHTTPBodyPart *bodyPart = [[AFHTTPBodyPart alloc] init];
  551. bodyPart.stringEncoding = self.stringEncoding;
  552. bodyPart.headers = headers;
  553. bodyPart.boundary = self.boundary;
  554. bodyPart.bodyContentLength = [body length];
  555. bodyPart.body = body;
  556. [self.bodyStream appendHTTPBodyPart:bodyPart];
  557. }
  558. - (void)throttleBandwidthWithPacketSize:(NSUInteger)numberOfBytes
  559. delay:(NSTimeInterval)delay
  560. {
  561. self.bodyStream.numberOfBytesInPacket = numberOfBytes;
  562. self.bodyStream.delay = delay;
  563. }
  564. - (NSMutableURLRequest *)requestByFinalizingMultipartFormData {
  565. if ([self.bodyStream isEmpty]) {
  566. return self.request;
  567. }
  568. // Reset the initial and final boundaries to ensure correct Content-Length
  569. [self.bodyStream setInitialAndFinalBoundaries];
  570. [self.request setHTTPBodyStream:self.bodyStream];
  571. [self.request setValue:[NSString stringWithFormat:@"multipart/form-data; boundary=%@", self.boundary] forHTTPHeaderField:@"Content-Type"];
  572. [self.request setValue:[NSString stringWithFormat:@"%llu", [self.bodyStream contentLength]] forHTTPHeaderField:@"Content-Length"];
  573. return self.request;
  574. }
  575. @end
  576. #pragma mark -
  577. @interface NSStream ()
  578. @property (readwrite) NSStreamStatus streamStatus;
  579. @property (readwrite, copy) NSError *streamError;
  580. @end
  581. @interface AFMultipartBodyStream () <NSCopying>
  582. @property (readwrite, nonatomic, assign) NSStringEncoding stringEncoding;
  583. @property (readwrite, nonatomic, strong) NSMutableArray *HTTPBodyParts;
  584. @property (readwrite, nonatomic, strong) NSEnumerator *HTTPBodyPartEnumerator;
  585. @property (readwrite, nonatomic, strong) AFHTTPBodyPart *currentHTTPBodyPart;
  586. @property (readwrite, nonatomic, strong) NSOutputStream *outputStream;
  587. @property (readwrite, nonatomic, strong) NSMutableData *buffer;
  588. @end
  589. @implementation AFMultipartBodyStream
  590. #pragma clang diagnostic push
  591. #pragma clang diagnostic ignored "-Wimplicit-atomic-properties"
  592. @synthesize streamStatus;
  593. @synthesize streamError;
  594. #pragma clang diagnostic pop
  595. - (id)initWithStringEncoding:(NSStringEncoding)encoding {
  596. self = [super init];
  597. if (!self) {
  598. return nil;
  599. }
  600. self.stringEncoding = encoding;
  601. self.HTTPBodyParts = [NSMutableArray array];
  602. self.numberOfBytesInPacket = NSIntegerMax;
  603. return self;
  604. }
  605. - (void)setInitialAndFinalBoundaries {
  606. if ([self.HTTPBodyParts count] > 0) {
  607. for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts) {
  608. bodyPart.hasInitialBoundary = NO;
  609. bodyPart.hasFinalBoundary = NO;
  610. }
  611. [[self.HTTPBodyParts objectAtIndex:0] setHasInitialBoundary:YES];
  612. [[self.HTTPBodyParts lastObject] setHasFinalBoundary:YES];
  613. }
  614. }
  615. - (void)appendHTTPBodyPart:(AFHTTPBodyPart *)bodyPart {
  616. [self.HTTPBodyParts addObject:bodyPart];
  617. }
  618. - (BOOL)isEmpty {
  619. return [self.HTTPBodyParts count] == 0;
  620. }
  621. #pragma mark - NSInputStream
  622. - (NSInteger)read:(uint8_t *)buffer
  623. maxLength:(NSUInteger)length
  624. {
  625. if ([self streamStatus] == NSStreamStatusClosed) {
  626. return 0;
  627. }
  628. NSInteger totalNumberOfBytesRead = 0;
  629. #pragma clang diagnostic push
  630. #pragma clang diagnostic ignored "-Wgnu"
  631. while ((NSUInteger)totalNumberOfBytesRead < MIN(length, self.numberOfBytesInPacket)) {
  632. if (!self.currentHTTPBodyPart || ![self.currentHTTPBodyPart hasBytesAvailable]) {
  633. if (!(self.currentHTTPBodyPart = [self.HTTPBodyPartEnumerator nextObject])) {
  634. break;
  635. }
  636. } else {
  637. NSUInteger maxLength = length - (NSUInteger)totalNumberOfBytesRead;
  638. NSInteger numberOfBytesRead = [self.currentHTTPBodyPart read:&buffer[totalNumberOfBytesRead] maxLength:maxLength];
  639. if (numberOfBytesRead == -1) {
  640. self.streamError = self.currentHTTPBodyPart.inputStream.streamError;
  641. break;
  642. } else {
  643. totalNumberOfBytesRead += numberOfBytesRead;
  644. if (self.delay > 0.0f) {
  645. [NSThread sleepForTimeInterval:self.delay];
  646. }
  647. }
  648. }
  649. }
  650. #pragma clang diagnostic pop
  651. return totalNumberOfBytesRead;
  652. }
  653. - (BOOL)getBuffer:(__unused uint8_t **)buffer
  654. length:(__unused NSUInteger *)len
  655. {
  656. return NO;
  657. }
  658. - (BOOL)hasBytesAvailable {
  659. return [self streamStatus] == NSStreamStatusOpen;
  660. }
  661. #pragma mark - NSStream
  662. - (void)open {
  663. if (self.streamStatus == NSStreamStatusOpen) {
  664. return;
  665. }
  666. self.streamStatus = NSStreamStatusOpen;
  667. [self setInitialAndFinalBoundaries];
  668. self.HTTPBodyPartEnumerator = [self.HTTPBodyParts objectEnumerator];
  669. }
  670. - (void)close {
  671. self.streamStatus = NSStreamStatusClosed;
  672. }
  673. - (id)propertyForKey:(__unused NSString *)key {
  674. return nil;
  675. }
  676. - (BOOL)setProperty:(__unused id)property
  677. forKey:(__unused NSString *)key
  678. {
  679. return NO;
  680. }
  681. - (void)scheduleInRunLoop:(__unused NSRunLoop *)aRunLoop
  682. forMode:(__unused NSString *)mode
  683. {}
  684. - (void)removeFromRunLoop:(__unused NSRunLoop *)aRunLoop
  685. forMode:(__unused NSString *)mode
  686. {}
  687. - (unsigned long long)contentLength {
  688. unsigned long long length = 0;
  689. for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts) {
  690. length += [bodyPart contentLength];
  691. }
  692. return length;
  693. }
  694. #pragma mark - Undocumented CFReadStream Bridged Methods
  695. - (void)_scheduleInCFRunLoop:(__unused CFRunLoopRef)aRunLoop
  696. forMode:(__unused CFStringRef)aMode
  697. {}
  698. - (void)_unscheduleFromCFRunLoop:(__unused CFRunLoopRef)aRunLoop
  699. forMode:(__unused CFStringRef)aMode
  700. {}
  701. - (BOOL)_setCFClientFlags:(__unused CFOptionFlags)inFlags
  702. callback:(__unused CFReadStreamClientCallBack)inCallback
  703. context:(__unused CFStreamClientContext *)inContext {
  704. return NO;
  705. }
  706. #pragma mark - NSCopying
  707. -(id)copyWithZone:(NSZone *)zone {
  708. AFMultipartBodyStream *bodyStreamCopy = [[[self class] allocWithZone:zone] initWithStringEncoding:self.stringEncoding];
  709. for (AFHTTPBodyPart *bodyPart in self.HTTPBodyParts) {
  710. [bodyStreamCopy appendHTTPBodyPart:[bodyPart copy]];
  711. }
  712. [bodyStreamCopy setInitialAndFinalBoundaries];
  713. return bodyStreamCopy;
  714. }
  715. @end
  716. #pragma mark -
  717. typedef enum {
  718. AFEncapsulationBoundaryPhase = 1,
  719. AFHeaderPhase = 2,
  720. AFBodyPhase = 3,
  721. AFFinalBoundaryPhase = 4,
  722. } AFHTTPBodyPartReadPhase;
  723. @interface AFHTTPBodyPart () <NSCopying> {
  724. AFHTTPBodyPartReadPhase _phase;
  725. NSInputStream *_inputStream;
  726. unsigned long long _phaseReadOffset;
  727. }
  728. - (BOOL)transitionToNextPhase;
  729. - (NSInteger)readData:(NSData *)data
  730. intoBuffer:(uint8_t *)buffer
  731. maxLength:(NSUInteger)length;
  732. @end
  733. @implementation AFHTTPBodyPart
  734. - (id)init {
  735. self = [super init];
  736. if (!self) {
  737. return nil;
  738. }
  739. [self transitionToNextPhase];
  740. return self;
  741. }
  742. - (void)dealloc {
  743. if (_inputStream) {
  744. [_inputStream close];
  745. _inputStream = nil;
  746. }
  747. }
  748. - (NSInputStream *)inputStream {
  749. if (!_inputStream) {
  750. if ([self.body isKindOfClass:[NSData class]]) {
  751. _inputStream = [NSInputStream inputStreamWithData:self.body];
  752. } else if ([self.body isKindOfClass:[NSURL class]]) {
  753. _inputStream = [NSInputStream inputStreamWithURL:self.body];
  754. } else if ([self.body isKindOfClass:[NSInputStream class]]) {
  755. _inputStream = self.body;
  756. }
  757. }
  758. return _inputStream;
  759. }
  760. - (NSString *)stringForHeaders {
  761. NSMutableString *headerString = [NSMutableString string];
  762. for (NSString *field in [self.headers allKeys]) {
  763. [headerString appendString:[NSString stringWithFormat:@"%@: %@%@", field, [self.headers valueForKey:field], kAFMultipartFormCRLF]];
  764. }
  765. [headerString appendString:kAFMultipartFormCRLF];
  766. return [NSString stringWithString:headerString];
  767. }
  768. - (unsigned long long)contentLength {
  769. unsigned long long length = 0;
  770. NSData *encapsulationBoundaryData = [([self hasInitialBoundary] ? AFMultipartFormInitialBoundary(self.boundary) : AFMultipartFormEncapsulationBoundary(self.boundary)) dataUsingEncoding:self.stringEncoding];
  771. length += [encapsulationBoundaryData length];
  772. NSData *headersData = [[self stringForHeaders] dataUsingEncoding:self.stringEncoding];
  773. length += [headersData length];
  774. length += _bodyContentLength;
  775. NSData *closingBoundaryData = ([self hasFinalBoundary] ? [AFMultipartFormFinalBoundary(self.boundary) dataUsingEncoding:self.stringEncoding] : [NSData data]);
  776. length += [closingBoundaryData length];
  777. return length;
  778. }
  779. - (BOOL)hasBytesAvailable {
  780. // Allows `read:maxLength:` to be called again if `AFMultipartFormFinalBoundary` doesn't fit into the available buffer
  781. if (_phase == AFFinalBoundaryPhase) {
  782. return YES;
  783. }
  784. #pragma clang diagnostic push
  785. #pragma clang diagnostic ignored "-Wcovered-switch-default"
  786. switch (self.inputStream.streamStatus) {
  787. case NSStreamStatusNotOpen:
  788. case NSStreamStatusOpening:
  789. case NSStreamStatusOpen:
  790. case NSStreamStatusReading:
  791. case NSStreamStatusWriting:
  792. return YES;
  793. case NSStreamStatusAtEnd:
  794. case NSStreamStatusClosed:
  795. case NSStreamStatusError:
  796. default:
  797. return NO;
  798. }
  799. #pragma clang diagnostic pop
  800. }
  801. - (NSInteger)read:(uint8_t *)buffer
  802. maxLength:(NSUInteger)length
  803. {
  804. NSInteger totalNumberOfBytesRead = 0;
  805. if (_phase == AFEncapsulationBoundaryPhase) {
  806. NSData *encapsulationBoundaryData = [([self hasInitialBoundary] ? AFMultipartFormInitialBoundary(self.boundary) : AFMultipartFormEncapsulationBoundary(self.boundary)) dataUsingEncoding:self.stringEncoding];
  807. totalNumberOfBytesRead += [self readData:encapsulationBoundaryData intoBuffer:&buffer[totalNumberOfBytesRead] maxLength:(length - (NSUInteger)totalNumberOfBytesRead)];
  808. }
  809. if (_phase == AFHeaderPhase) {
  810. NSData *headersData = [[self stringForHeaders] dataUsingEncoding:self.stringEncoding];
  811. totalNumberOfBytesRead += [self readData:headersData intoBuffer:&buffer[totalNumberOfBytesRead] maxLength:(length - (NSUInteger)totalNumberOfBytesRead)];
  812. }
  813. if (_phase == AFBodyPhase) {
  814. NSInteger numberOfBytesRead = 0;
  815. numberOfBytesRead = [self.inputStream read:&buffer[totalNumberOfBytesRead] maxLength:(length - (NSUInteger)totalNumberOfBytesRead)];
  816. if (numberOfBytesRead == -1) {
  817. return -1;
  818. } else {
  819. totalNumberOfBytesRead += numberOfBytesRead;
  820. if ([self.inputStream streamStatus] >= NSStreamStatusAtEnd) {
  821. [self transitionToNextPhase];
  822. }
  823. }
  824. }
  825. if (_phase == AFFinalBoundaryPhase) {
  826. NSData *closingBoundaryData = ([self hasFinalBoundary] ? [AFMultipartFormFinalBoundary(self.boundary) dataUsingEncoding:self.stringEncoding] : [NSData data]);
  827. totalNumberOfBytesRead += [self readData:closingBoundaryData intoBuffer:&buffer[totalNumberOfBytesRead] maxLength:(length - (NSUInteger)totalNumberOfBytesRead)];
  828. }
  829. return totalNumberOfBytesRead;
  830. }
  831. - (NSInteger)readData:(NSData *)data
  832. intoBuffer:(uint8_t *)buffer
  833. maxLength:(NSUInteger)length
  834. {
  835. #pragma clang diagnostic push
  836. #pragma clang diagnostic ignored "-Wgnu"
  837. NSRange range = NSMakeRange((NSUInteger)_phaseReadOffset, MIN([data length] - ((NSUInteger)_phaseReadOffset), length));
  838. [data getBytes:buffer range:range];
  839. #pragma clang diagnostic pop
  840. _phaseReadOffset += range.length;
  841. if (((NSUInteger)_phaseReadOffset) >= [data length]) {
  842. [self transitionToNextPhase];
  843. }
  844. return (NSInteger)range.length;
  845. }
  846. - (BOOL)transitionToNextPhase {
  847. if (![[NSThread currentThread] isMainThread]) {
  848. [self performSelectorOnMainThread:@selector(transitionToNextPhase) withObject:nil waitUntilDone:YES];
  849. return YES;
  850. }
  851. #pragma clang diagnostic push
  852. #pragma clang diagnostic ignored "-Wcovered-switch-default"
  853. switch (_phase) {
  854. case AFEncapsulationBoundaryPhase:
  855. _phase = AFHeaderPhase;
  856. break;
  857. case AFHeaderPhase:
  858. [self.inputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSRunLoopCommonModes];
  859. [self.inputStream open];
  860. _phase = AFBodyPhase;
  861. break;
  862. case AFBodyPhase:
  863. [self.inputStream close];
  864. _phase = AFFinalBoundaryPhase;
  865. break;
  866. case AFFinalBoundaryPhase:
  867. default:
  868. _phase = AFEncapsulationBoundaryPhase;
  869. break;
  870. }
  871. _phaseReadOffset = 0;
  872. #pragma clang diagnostic pop
  873. return YES;
  874. }
  875. #pragma mark - NSCopying
  876. - (id)copyWithZone:(NSZone *)zone {
  877. AFHTTPBodyPart *bodyPart = [[[self class] allocWithZone:zone] init];
  878. bodyPart.stringEncoding = self.stringEncoding;
  879. bodyPart.headers = self.headers;
  880. bodyPart.bodyContentLength = self.bodyContentLength;
  881. bodyPart.body = self.body;
  882. bodyPart.boundary = self.boundary;
  883. return bodyPart;
  884. }
  885. @end
  886. #pragma mark -
  887. @implementation AFJSONRequestSerializer
  888. + (instancetype)serializer {
  889. return [self serializerWithWritingOptions:0];
  890. }
  891. + (instancetype)serializerWithWritingOptions:(NSJSONWritingOptions)writingOptions
  892. {
  893. AFJSONRequestSerializer *serializer = [[self alloc] init];
  894. serializer.writingOptions = writingOptions;
  895. return serializer;
  896. }
  897. #pragma mark - AFURLRequestSerialization
  898. - (NSURLRequest *)requestBySerializingRequest:(NSURLRequest *)request
  899. withParameters:(id)parameters
  900. error:(NSError *__autoreleasing *)error
  901. {
  902. NSParameterAssert(request);
  903. if ([self.HTTPMethodsEncodingParametersInURI containsObject:[[request HTTPMethod] uppercaseString]]) {
  904. return [super requestBySerializingRequest:request withParameters:parameters error:error];
  905. }
  906. NSMutableURLRequest *mutableRequest = [request mutableCopy];
  907. [self.HTTPRequestHeaders enumerateKeysAndObjectsUsingBlock:^(id field, id value, BOOL * __unused stop) {
  908. if (![request valueForHTTPHeaderField:field]) {
  909. [mutableRequest setValue:value forHTTPHeaderField:field];
  910. }
  911. }];
  912. if (parameters) {
  913. if (![mutableRequest valueForHTTPHeaderField:@"Content-Type"]) {
  914. NSString *charset = (__bridge NSString *)CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(NSUTF8StringEncoding));
  915. [mutableRequest setValue:[NSString stringWithFormat:@"application/json; charset=%@", charset] forHTTPHeaderField:@"Content-Type"];
  916. }
  917. [mutableRequest setHTTPBody:[NSJSONSerialization dataWithJSONObject:parameters options:self.writingOptions error:error]];
  918. }
  919. return mutableRequest;
  920. }
  921. #pragma mark - NSecureCoding
  922. - (id)initWithCoder:(NSCoder *)decoder {
  923. self = [super initWithCoder:decoder];
  924. if (!self) {
  925. return nil;
  926. }
  927. self.writingOptions = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(writingOptions))] unsignedIntegerValue];
  928. return self;
  929. }
  930. - (void)encodeWithCoder:(NSCoder *)coder {
  931. [super encodeWithCoder:coder];
  932. [coder encodeInteger:self.writingOptions forKey:NSStringFromSelector(@selector(writingOptions))];
  933. }
  934. #pragma mark - NSCopying
  935. - (id)copyWithZone:(NSZone *)zone {
  936. AFJSONRequestSerializer *serializer = [super copyWithZone:zone];
  937. serializer.writingOptions = self.writingOptions;
  938. return serializer;
  939. }
  940. @end
  941. #pragma mark -
  942. @implementation AFPropertyListRequestSerializer
  943. + (instancetype)serializer {
  944. return [self serializerWithFormat:NSPropertyListXMLFormat_v1_0 writeOptions:0];
  945. }
  946. + (instancetype)serializerWithFormat:(NSPropertyListFormat)format
  947. writeOptions:(NSPropertyListWriteOptions)writeOptions
  948. {
  949. AFPropertyListRequestSerializer *serializer = [[self alloc] init];
  950. serializer.format = format;
  951. serializer.writeOptions = writeOptions;
  952. return serializer;
  953. }
  954. #pragma mark - AFURLRequestSerializer
  955. - (NSURLRequest *)requestBySerializingRequest:(NSURLRequest *)request
  956. withParameters:(id)parameters
  957. error:(NSError *__autoreleasing *)error
  958. {
  959. NSParameterAssert(request);
  960. if ([self.HTTPMethodsEncodingParametersInURI containsObject:[[request HTTPMethod] uppercaseString]]) {
  961. return [super requestBySerializingRequest:request withParameters:parameters error:error];
  962. }
  963. NSMutableURLRequest *mutableRequest = [request mutableCopy];
  964. [self.HTTPRequestHeaders enumerateKeysAndObjectsUsingBlock:^(id field, id value, BOOL * __unused stop) {
  965. if (![request valueForHTTPHeaderField:field]) {
  966. [mutableRequest setValue:value forHTTPHeaderField:field];
  967. }
  968. }];
  969. if (parameters) {
  970. if (![mutableRequest valueForHTTPHeaderField:@"Content-Type"]) {
  971. NSString *charset = (__bridge NSString *)CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(NSUTF8StringEncoding));
  972. [mutableRequest setValue:[NSString stringWithFormat:@"application/x-plist; charset=%@", charset] forHTTPHeaderField:@"Content-Type"];
  973. }
  974. [mutableRequest setHTTPBody:[NSPropertyListSerialization dataWithPropertyList:parameters format:self.format options:self.writeOptions error:error]];
  975. }
  976. return mutableRequest;
  977. }
  978. #pragma mark - NSecureCoding
  979. - (id)initWithCoder:(NSCoder *)decoder {
  980. self = [super initWithCoder:decoder];
  981. if (!self) {
  982. return nil;
  983. }
  984. self.format = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(format))] unsignedIntegerValue];
  985. self.writeOptions = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(writeOptions))] unsignedIntegerValue];
  986. return self;
  987. }
  988. - (void)encodeWithCoder:(NSCoder *)coder {
  989. [super encodeWithCoder:coder];
  990. [coder encodeInteger:self.format forKey:NSStringFromSelector(@selector(format))];
  991. [coder encodeObject:@(self.writeOptions) forKey:NSStringFromSelector(@selector(writeOptions))];
  992. }
  993. #pragma mark - NSCopying
  994. - (id)copyWithZone:(NSZone *)zone {
  995. AFPropertyListRequestSerializer *serializer = [super copyWithZone:zone];
  996. serializer.format = self.format;
  997. serializer.writeOptions = self.writeOptions;
  998. return serializer;
  999. }
  1000. @end