AFHTTPSessionManager.m 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361
  1. // AFHTTPSessionManager.m
  2. // Copyright (c) 2011–2016 Alamofire Software Foundation ( http://alamofire.org/ )
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to deal
  6. // in the Software without restriction, including without limitation the rights
  7. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. // copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20. // THE SOFTWARE.
  21. #import "AFHTTPSessionManager.h"
  22. #import "AFURLRequestSerialization.h"
  23. #import "AFURLResponseSerialization.h"
  24. #import <Availability.h>
  25. #import <TargetConditionals.h>
  26. #import <Security/Security.h>
  27. #import <netinet/in.h>
  28. #import <netinet6/in6.h>
  29. #import <arpa/inet.h>
  30. #import <ifaddrs.h>
  31. #import <netdb.h>
  32. #if TARGET_OS_IOS || TARGET_OS_TV
  33. #import <UIKit/UIKit.h>
  34. #elif TARGET_OS_WATCH
  35. #import <WatchKit/WatchKit.h>
  36. #endif
  37. @interface AFHTTPSessionManager ()
  38. @property (readwrite, nonatomic, strong) NSURL *baseURL;
  39. @end
  40. @implementation AFHTTPSessionManager
  41. @dynamic responseSerializer;
  42. + (instancetype)manager {
  43. return [[[self class] alloc] initWithBaseURL:nil];
  44. }
  45. - (instancetype)init {
  46. return [self initWithBaseURL:nil];
  47. }
  48. - (instancetype)initWithBaseURL:(NSURL *)url {
  49. return [self initWithBaseURL:url sessionConfiguration:nil];
  50. }
  51. - (instancetype)initWithSessionConfiguration:(NSURLSessionConfiguration *)configuration {
  52. return [self initWithBaseURL:nil sessionConfiguration:configuration];
  53. }
  54. - (instancetype)initWithBaseURL:(NSURL *)url
  55. sessionConfiguration:(NSURLSessionConfiguration *)configuration
  56. {
  57. self = [super initWithSessionConfiguration:configuration];
  58. if (!self) {
  59. return nil;
  60. }
  61. // Ensure terminal slash for baseURL path, so that NSURL +URLWithString:relativeToURL: works as expected
  62. if ([[url path] length] > 0 && ![[url absoluteString] hasSuffix:@"/"]) {
  63. url = [url URLByAppendingPathComponent:@""];
  64. }
  65. self.baseURL = url;
  66. self.requestSerializer = [AFHTTPRequestSerializer serializer];
  67. self.responseSerializer = [AFJSONResponseSerializer serializer];
  68. return self;
  69. }
  70. #pragma mark -
  71. - (void)setRequestSerializer:(AFHTTPRequestSerializer <AFURLRequestSerialization> *)requestSerializer {
  72. NSParameterAssert(requestSerializer);
  73. _requestSerializer = requestSerializer;
  74. }
  75. - (void)setResponseSerializer:(AFHTTPResponseSerializer <AFURLResponseSerialization> *)responseSerializer {
  76. NSParameterAssert(responseSerializer);
  77. [super setResponseSerializer:responseSerializer];
  78. }
  79. #pragma mark -
  80. - (NSURLSessionDataTask *)GET:(NSString *)URLString
  81. parameters:(id)parameters
  82. success:(void (^)(NSURLSessionDataTask *task, id responseObject))success
  83. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  84. {
  85. return [self GET:URLString parameters:parameters progress:nil success:success failure:failure];
  86. }
  87. - (NSURLSessionDataTask *)GET:(NSString *)URLString
  88. parameters:(id)parameters
  89. progress:(void (^)(NSProgress * _Nonnull))downloadProgress
  90. success:(void (^)(NSURLSessionDataTask * _Nonnull, id _Nullable))success
  91. failure:(void (^)(NSURLSessionDataTask * _Nullable, NSError * _Nonnull))failure
  92. {
  93. NSURLSessionDataTask *dataTask = [self dataTaskWithHTTPMethod:@"GET"
  94. URLString:URLString
  95. parameters:parameters
  96. uploadProgress:nil
  97. downloadProgress:downloadProgress
  98. success:success
  99. failure:failure];
  100. [dataTask resume];
  101. return dataTask;
  102. }
  103. - (NSURLSessionDataTask *)HEAD:(NSString *)URLString
  104. parameters:(id)parameters
  105. success:(void (^)(NSURLSessionDataTask *task))success
  106. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  107. {
  108. NSURLSessionDataTask *dataTask = [self dataTaskWithHTTPMethod:@"HEAD" URLString:URLString parameters:parameters uploadProgress:nil downloadProgress:nil success:^(NSURLSessionDataTask *task, __unused id responseObject) {
  109. if (success) {
  110. success(task);
  111. }
  112. } failure:failure];
  113. [dataTask resume];
  114. return dataTask;
  115. }
  116. - (NSURLSessionDataTask *)POST:(NSString *)URLString
  117. parameters:(id)parameters
  118. success:(void (^)(NSURLSessionDataTask *task, id responseObject))success
  119. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  120. {
  121. return [self POST:URLString parameters:parameters progress:nil success:success failure:failure];
  122. }
  123. - (NSURLSessionDataTask *)POST:(NSString *)URLString
  124. parameters:(id)parameters
  125. progress:(void (^)(NSProgress * _Nonnull))uploadProgress
  126. success:(void (^)(NSURLSessionDataTask * _Nonnull, id _Nullable))success
  127. failure:(void (^)(NSURLSessionDataTask * _Nullable, NSError * _Nonnull))failure
  128. {
  129. NSURLSessionDataTask *dataTask = [self dataTaskWithHTTPMethod:@"POST" URLString:URLString parameters:parameters uploadProgress:uploadProgress downloadProgress:nil success:success failure:failure];
  130. [dataTask resume];
  131. return dataTask;
  132. }
  133. - (NSURLSessionDataTask *)POST:(NSString *)URLString
  134. parameters:(nullable id)parameters
  135. constructingBodyWithBlock:(nullable void (^)(id<AFMultipartFormData> _Nonnull))block
  136. success:(nullable void (^)(NSURLSessionDataTask * _Nonnull, id _Nullable))success
  137. failure:(nullable void (^)(NSURLSessionDataTask * _Nullable, NSError * _Nonnull))failure
  138. {
  139. return [self POST:URLString parameters:parameters constructingBodyWithBlock:block progress:nil success:success failure:failure];
  140. }
  141. - (NSURLSessionDataTask *)POST:(NSString *)URLString
  142. parameters:(id)parameters
  143. constructingBodyWithBlock:(void (^)(id <AFMultipartFormData> formData))block
  144. progress:(nullable void (^)(NSProgress * _Nonnull))uploadProgress
  145. success:(void (^)(NSURLSessionDataTask *task, id responseObject))success
  146. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  147. {
  148. NSError *serializationError = nil;
  149. NSMutableURLRequest *request = [self.requestSerializer multipartFormRequestWithMethod:@"POST" URLString:[[NSURL URLWithString:URLString relativeToURL:self.baseURL] absoluteString] parameters:parameters constructingBodyWithBlock:block error:&serializationError];
  150. if (serializationError) {
  151. if (failure) {
  152. #pragma clang diagnostic push
  153. #pragma clang diagnostic ignored "-Wgnu"
  154. dispatch_async(self.completionQueue ?: dispatch_get_main_queue(), ^{
  155. failure(nil, serializationError);
  156. });
  157. #pragma clang diagnostic pop
  158. }
  159. return nil;
  160. }
  161. __block NSURLSessionDataTask *task = [self uploadTaskWithStreamedRequest:request progress:uploadProgress completionHandler:^(NSURLResponse * __unused response, id responseObject, NSError *error) {
  162. if (error) {
  163. if (failure) {
  164. failure(task, error);
  165. }
  166. } else {
  167. if (success) {
  168. success(task, responseObject);
  169. }
  170. }
  171. }];
  172. [task resume];
  173. return task;
  174. }
  175. - (NSURLSessionDataTask *)PUT:(NSString *)URLString
  176. parameters:(id)parameters
  177. success:(void (^)(NSURLSessionDataTask *task, id responseObject))success
  178. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  179. {
  180. NSURLSessionDataTask *dataTask = [self dataTaskWithHTTPMethod:@"PUT" URLString:URLString parameters:parameters uploadProgress:nil downloadProgress:nil success:success failure:failure];
  181. [dataTask resume];
  182. return dataTask;
  183. }
  184. - (NSURLSessionDataTask *)PATCH:(NSString *)URLString
  185. parameters:(id)parameters
  186. success:(void (^)(NSURLSessionDataTask *task, id responseObject))success
  187. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  188. {
  189. NSURLSessionDataTask *dataTask = [self dataTaskWithHTTPMethod:@"PATCH" URLString:URLString parameters:parameters uploadProgress:nil downloadProgress:nil success:success failure:failure];
  190. [dataTask resume];
  191. return dataTask;
  192. }
  193. - (NSURLSessionDataTask *)DELETE:(NSString *)URLString
  194. parameters:(id)parameters
  195. success:(void (^)(NSURLSessionDataTask *task, id responseObject))success
  196. failure:(void (^)(NSURLSessionDataTask *task, NSError *error))failure
  197. {
  198. NSURLSessionDataTask *dataTask = [self dataTaskWithHTTPMethod:@"DELETE" URLString:URLString parameters:parameters uploadProgress:nil downloadProgress:nil success:success failure:failure];
  199. [dataTask resume];
  200. return dataTask;
  201. }
  202. - (NSURLSessionDataTask *)dataTaskWithHTTPMethod:(NSString *)method
  203. URLString:(NSString *)URLString
  204. parameters:(id)parameters
  205. uploadProgress:(nullable void (^)(NSProgress *uploadProgress)) uploadProgress
  206. downloadProgress:(nullable void (^)(NSProgress *downloadProgress)) downloadProgress
  207. success:(void (^)(NSURLSessionDataTask *, id))success
  208. failure:(void (^)(NSURLSessionDataTask *, NSError *))failure
  209. {
  210. NSError *serializationError = nil;
  211. NSMutableURLRequest *request = [self.requestSerializer requestWithMethod:method URLString:[[NSURL URLWithString:URLString relativeToURL:self.baseURL] absoluteString] parameters:parameters error:&serializationError];
  212. if (serializationError) {
  213. if (failure) {
  214. #pragma clang diagnostic push
  215. #pragma clang diagnostic ignored "-Wgnu"
  216. dispatch_async(self.completionQueue ?: dispatch_get_main_queue(), ^{
  217. failure(nil, serializationError);
  218. });
  219. #pragma clang diagnostic pop
  220. }
  221. return nil;
  222. }
  223. __block NSURLSessionDataTask *dataTask = nil;
  224. dataTask = [self dataTaskWithRequest:request
  225. uploadProgress:uploadProgress
  226. downloadProgress:downloadProgress
  227. completionHandler:^(NSURLResponse * __unused response, id responseObject, NSError *error) {
  228. if (error) {
  229. if (failure) {
  230. failure(dataTask, error);
  231. }
  232. } else {
  233. if (success) {
  234. success(dataTask, responseObject);
  235. }
  236. }
  237. }];
  238. return dataTask;
  239. }
  240. #pragma mark - NSObject
  241. - (NSString *)description {
  242. return [NSString stringWithFormat:@"<%@: %p, baseURL: %@, session: %@, operationQueue: %@>", NSStringFromClass([self class]), self, [self.baseURL absoluteString], self.session, self.operationQueue];
  243. }
  244. #pragma mark - NSSecureCoding
  245. + (BOOL)supportsSecureCoding {
  246. return YES;
  247. }
  248. - (instancetype)initWithCoder:(NSCoder *)decoder {
  249. NSURL *baseURL = [decoder decodeObjectOfClass:[NSURL class] forKey:NSStringFromSelector(@selector(baseURL))];
  250. NSURLSessionConfiguration *configuration = [decoder decodeObjectOfClass:[NSURLSessionConfiguration class] forKey:@"sessionConfiguration"];
  251. if (!configuration) {
  252. NSString *configurationIdentifier = [decoder decodeObjectOfClass:[NSString class] forKey:@"identifier"];
  253. if (configurationIdentifier) {
  254. #if (defined(__IPHONE_OS_VERSION_MIN_REQUIRED) && __IPHONE_OS_VERSION_MIN_REQUIRED >= 80000) || (defined(__MAC_OS_X_VERSION_MIN_REQUIRED) && __MAC_OS_X_VERSION_MIN_REQUIRED >= 1100)
  255. configuration = [NSURLSessionConfiguration backgroundSessionConfigurationWithIdentifier:configurationIdentifier];
  256. #else
  257. configuration = [NSURLSessionConfiguration backgroundSessionConfiguration:configurationIdentifier];
  258. #endif
  259. }
  260. }
  261. self = [self initWithBaseURL:baseURL sessionConfiguration:configuration];
  262. if (!self) {
  263. return nil;
  264. }
  265. self.requestSerializer = [decoder decodeObjectOfClass:[AFHTTPRequestSerializer class] forKey:NSStringFromSelector(@selector(requestSerializer))];
  266. self.responseSerializer = [decoder decodeObjectOfClass:[AFHTTPResponseSerializer class] forKey:NSStringFromSelector(@selector(responseSerializer))];
  267. AFSecurityPolicy *decodedPolicy = [decoder decodeObjectOfClass:[AFSecurityPolicy class] forKey:NSStringFromSelector(@selector(securityPolicy))];
  268. if (decodedPolicy) {
  269. self.securityPolicy = decodedPolicy;
  270. }
  271. return self;
  272. }
  273. - (void)encodeWithCoder:(NSCoder *)coder {
  274. [super encodeWithCoder:coder];
  275. [coder encodeObject:self.baseURL forKey:NSStringFromSelector(@selector(baseURL))];
  276. if ([self.session.configuration conformsToProtocol:@protocol(NSCoding)]) {
  277. [coder encodeObject:self.session.configuration forKey:@"sessionConfiguration"];
  278. } else {
  279. [coder encodeObject:self.session.configuration.identifier forKey:@"identifier"];
  280. }
  281. [coder encodeObject:self.requestSerializer forKey:NSStringFromSelector(@selector(requestSerializer))];
  282. [coder encodeObject:self.responseSerializer forKey:NSStringFromSelector(@selector(responseSerializer))];
  283. [coder encodeObject:self.securityPolicy forKey:NSStringFromSelector(@selector(securityPolicy))];
  284. }
  285. #pragma mark - NSCopying
  286. - (instancetype)copyWithZone:(NSZone *)zone {
  287. AFHTTPSessionManager *HTTPClient = [[[self class] allocWithZone:zone] initWithBaseURL:self.baseURL sessionConfiguration:self.session.configuration];
  288. HTTPClient.requestSerializer = [self.requestSerializer copyWithZone:zone];
  289. HTTPClient.responseSerializer = [self.responseSerializer copyWithZone:zone];
  290. HTTPClient.securityPolicy = [self.securityPolicy copyWithZone:zone];
  291. return HTTPClient;
  292. }
  293. @end