123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836 |
- // AFURLResponseSerialization.m
- // Copyright (c) 2011–2016 Alamofire Software Foundation ( http://alamofire.org/ )
- //
- // 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 "AFURLResponseSerialization.h"
- #import <TargetConditionals.h>
- #if TARGET_OS_IOS
- #import <UIKit/UIKit.h>
- #elif TARGET_OS_WATCH
- #import <WatchKit/WatchKit.h>
- #elif defined(__MAC_OS_X_VERSION_MIN_REQUIRED)
- #import <Cocoa/Cocoa.h>
- #endif
- NSString * const AFURLResponseSerializationErrorDomain = @"com.alamofire.error.serialization.response";
- NSString * const AFNetworkingOperationFailingURLResponseErrorKey = @"com.alamofire.serialization.response.error.response";
- NSString * const AFNetworkingOperationFailingURLResponseDataErrorKey = @"com.alamofire.serialization.response.error.data";
- static NSError * AFErrorWithUnderlyingError(NSError *error, NSError *underlyingError) {
- if (!error) {
- return underlyingError;
- }
- if (!underlyingError || error.userInfo[NSUnderlyingErrorKey]) {
- return error;
- }
- NSMutableDictionary *mutableUserInfo = [error.userInfo mutableCopy];
- mutableUserInfo[NSUnderlyingErrorKey] = underlyingError;
- return [[NSError alloc] initWithDomain:error.domain code:error.code userInfo:mutableUserInfo];
- }
- static BOOL AFErrorOrUnderlyingErrorHasCodeInDomain(NSError *error, NSInteger code, NSString *domain) {
- if ([error.domain isEqualToString:domain] && error.code == code) {
- return YES;
- } else if (error.userInfo[NSUnderlyingErrorKey]) {
- return AFErrorOrUnderlyingErrorHasCodeInDomain(error.userInfo[NSUnderlyingErrorKey], code, domain);
- }
- return NO;
- }
- id AFJSONObjectByRemovingKeysWithNullValues(id JSONObject, NSJSONReadingOptions readingOptions) {
- if ([JSONObject isKindOfClass:[NSArray class]]) {
- NSMutableArray *mutableArray = [NSMutableArray arrayWithCapacity:[(NSArray *)JSONObject count]];
- for (id value in (NSArray *)JSONObject) {
- if (![value isEqual:[NSNull null]]) {
- [mutableArray addObject:AFJSONObjectByRemovingKeysWithNullValues(value, readingOptions)];
- }
- }
- return (readingOptions & NSJSONReadingMutableContainers) ? mutableArray : [NSArray arrayWithArray:mutableArray];
- } else if ([JSONObject isKindOfClass:[NSDictionary class]]) {
- NSMutableDictionary *mutableDictionary = [NSMutableDictionary dictionaryWithDictionary:JSONObject];
- for (id <NSCopying> key in [(NSDictionary *)JSONObject allKeys]) {
- id value = (NSDictionary *)JSONObject[key];
- if (!value || [value isEqual:[NSNull null]]) {
- [mutableDictionary removeObjectForKey:key];
- } else if ([value isKindOfClass:[NSArray class]] || [value isKindOfClass:[NSDictionary class]]) {
- mutableDictionary[key] = AFJSONObjectByRemovingKeysWithNullValues(value, readingOptions);
- }
- }
- return (readingOptions & NSJSONReadingMutableContainers) ? mutableDictionary : [NSDictionary dictionaryWithDictionary:mutableDictionary];
- }
- return JSONObject;
- }
- @implementation AFHTTPResponseSerializer
- + (instancetype)serializer {
- return [[self alloc] init];
- }
- - (instancetype)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- self.acceptableStatusCodes = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(200, 100)];
- self.acceptableContentTypes = nil;
- return self;
- }
- #pragma mark -
- - (BOOL)validateResponse:(NSHTTPURLResponse *)response
- data:(NSData *)data
- error:(NSError * __autoreleasing *)error
- {
- BOOL responseIsValid = YES;
- NSError *validationError = nil;
- if ([response isKindOfClass:[NSHTTPURLResponse class]]) {
- if (self.acceptableContentTypes && ![self.acceptableContentTypes containsObject:[response MIMEType]] &&
- !([response MIMEType] == nil && [data length] == 0)) {
- if ([data length] > 0 && [response URL]) {
- NSMutableDictionary *mutableUserInfo = [@{
- NSLocalizedDescriptionKey: [NSString stringWithFormat:NSLocalizedStringFromTable(@"Request failed: unacceptable content-type: %@", @"AFNetworking", nil), [response MIMEType]],
- NSURLErrorFailingURLErrorKey:[response URL],
- AFNetworkingOperationFailingURLResponseErrorKey: response,
- } mutableCopy];
- if (data) {
- mutableUserInfo[AFNetworkingOperationFailingURLResponseDataErrorKey] = data;
- }
- validationError = AFErrorWithUnderlyingError([NSError errorWithDomain:AFURLResponseSerializationErrorDomain code:NSURLErrorCannotDecodeContentData userInfo:mutableUserInfo], validationError);
- }
- responseIsValid = NO;
- }
- if (self.acceptableStatusCodes && ![self.acceptableStatusCodes containsIndex:(NSUInteger)response.statusCode] && [response URL]) {
- NSMutableDictionary *mutableUserInfo = [@{
- NSLocalizedDescriptionKey: [NSString stringWithFormat:NSLocalizedStringFromTable(@"Request failed: %@ (%ld)", @"AFNetworking", nil), [NSHTTPURLResponse localizedStringForStatusCode:response.statusCode], (long)response.statusCode],
- NSURLErrorFailingURLErrorKey:[response URL],
- AFNetworkingOperationFailingURLResponseErrorKey: response,
- } mutableCopy];
- if (data) {
- mutableUserInfo[AFNetworkingOperationFailingURLResponseDataErrorKey] = data;
- }
- validationError = AFErrorWithUnderlyingError([NSError errorWithDomain:AFURLResponseSerializationErrorDomain code:NSURLErrorBadServerResponse userInfo:mutableUserInfo], validationError);
- responseIsValid = NO;
- }
- }
- if (error && !responseIsValid) {
- *error = validationError;
- }
- return responseIsValid;
- }
- #pragma mark - AFURLResponseSerialization
- - (id)responseObjectForResponse:(NSURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- [self validateResponse:(NSHTTPURLResponse *)response data:data error:error];
- return data;
- }
- #pragma mark - NSSecureCoding
- + (BOOL)supportsSecureCoding {
- return YES;
- }
- - (instancetype)initWithCoder:(NSCoder *)decoder {
- self = [self init];
- if (!self) {
- return nil;
- }
- self.acceptableStatusCodes = [decoder decodeObjectOfClass:[NSIndexSet class] forKey:NSStringFromSelector(@selector(acceptableStatusCodes))];
- self.acceptableContentTypes = [decoder decodeObjectOfClass:[NSSet class] forKey:NSStringFromSelector(@selector(acceptableContentTypes))];
- return self;
- }
- - (void)encodeWithCoder:(NSCoder *)coder {
- [coder encodeObject:self.acceptableStatusCodes forKey:NSStringFromSelector(@selector(acceptableStatusCodes))];
- [coder encodeObject:self.acceptableContentTypes forKey:NSStringFromSelector(@selector(acceptableContentTypes))];
- }
- #pragma mark - NSCopying
- - (instancetype)copyWithZone:(NSZone *)zone {
- AFHTTPResponseSerializer *serializer = [[[self class] allocWithZone:zone] init];
- serializer.acceptableStatusCodes = [self.acceptableStatusCodes copyWithZone:zone];
- serializer.acceptableContentTypes = [self.acceptableContentTypes copyWithZone:zone];
- return serializer;
- }
- @end
- #pragma mark -
- @implementation AFJSONResponseSerializer
- + (instancetype)serializer {
- return [self serializerWithReadingOptions:(NSJSONReadingOptions)0];
- }
- + (instancetype)serializerWithReadingOptions:(NSJSONReadingOptions)readingOptions {
- AFJSONResponseSerializer *serializer = [[self alloc] init];
- serializer.readingOptions = readingOptions;
- return serializer;
- }
- - (instancetype)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- self.acceptableContentTypes = [NSSet setWithObjects:@"application/json", @"text/json", @"text/javascript", nil];
- return self;
- }
- #pragma mark - AFURLResponseSerialization
- - (id)responseObjectForResponse:(NSURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- if (![self validateResponse:(NSHTTPURLResponse *)response data:data error:error]) {
- if (!error || AFErrorOrUnderlyingErrorHasCodeInDomain(*error, NSURLErrorCannotDecodeContentData, AFURLResponseSerializationErrorDomain)) {
- return nil;
- }
- }
- // Workaround for behavior of Rails to return a single space for `head :ok` (a workaround for a bug in Safari), which is not interpreted as valid input by NSJSONSerialization.
- // See https://github.com/rails/rails/issues/1742
- BOOL isSpace = [data isEqualToData:[NSData dataWithBytes:" " length:1]];
-
- if (data.length == 0 || isSpace) {
- return nil;
- }
-
- NSError *serializationError = nil;
-
- id responseObject = [NSJSONSerialization JSONObjectWithData:data options:self.readingOptions error:&serializationError];
- if (!responseObject)
- {
- if (error) {
- *error = AFErrorWithUnderlyingError(serializationError, *error);
- }
- return nil;
- }
-
- if (self.removesKeysWithNullValues) {
- return AFJSONObjectByRemovingKeysWithNullValues(responseObject, self.readingOptions);
- }
- return responseObject;
- }
- #pragma mark - NSSecureCoding
- + (BOOL)supportsSecureCoding {
- return YES;
- }
- - (instancetype)initWithCoder:(NSCoder *)decoder {
- self = [super initWithCoder:decoder];
- if (!self) {
- return nil;
- }
- self.readingOptions = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(readingOptions))] unsignedIntegerValue];
- self.removesKeysWithNullValues = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(removesKeysWithNullValues))] boolValue];
- return self;
- }
- - (void)encodeWithCoder:(NSCoder *)coder {
- [super encodeWithCoder:coder];
- [coder encodeObject:@(self.readingOptions) forKey:NSStringFromSelector(@selector(readingOptions))];
- [coder encodeObject:@(self.removesKeysWithNullValues) forKey:NSStringFromSelector(@selector(removesKeysWithNullValues))];
- }
- #pragma mark - NSCopying
- - (instancetype)copyWithZone:(NSZone *)zone {
- AFJSONResponseSerializer *serializer = [super copyWithZone:zone];
- serializer.readingOptions = self.readingOptions;
- serializer.removesKeysWithNullValues = self.removesKeysWithNullValues;
- return serializer;
- }
- @end
- #pragma mark -
- @implementation AFXMLParserResponseSerializer
- + (instancetype)serializer {
- AFXMLParserResponseSerializer *serializer = [[self alloc] init];
- return serializer;
- }
- - (instancetype)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- self.acceptableContentTypes = [[NSSet alloc] initWithObjects:@"application/xml", @"text/xml", nil];
- return self;
- }
- #pragma mark - AFURLResponseSerialization
- - (id)responseObjectForResponse:(NSHTTPURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- if (![self validateResponse:(NSHTTPURLResponse *)response data:data error:error]) {
- if (!error || AFErrorOrUnderlyingErrorHasCodeInDomain(*error, NSURLErrorCannotDecodeContentData, AFURLResponseSerializationErrorDomain)) {
- return nil;
- }
- }
- return [[NSXMLParser alloc] initWithData:data];
- }
- @end
- #pragma mark -
- #ifdef __MAC_OS_X_VERSION_MIN_REQUIRED
- @implementation AFXMLDocumentResponseSerializer
- + (instancetype)serializer {
- return [self serializerWithXMLDocumentOptions:0];
- }
- + (instancetype)serializerWithXMLDocumentOptions:(NSUInteger)mask {
- AFXMLDocumentResponseSerializer *serializer = [[self alloc] init];
- serializer.options = mask;
- return serializer;
- }
- - (instancetype)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- self.acceptableContentTypes = [[NSSet alloc] initWithObjects:@"application/xml", @"text/xml", nil];
- return self;
- }
- #pragma mark - AFURLResponseSerialization
- - (id)responseObjectForResponse:(NSURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- if (![self validateResponse:(NSHTTPURLResponse *)response data:data error:error]) {
- if (!error || AFErrorOrUnderlyingErrorHasCodeInDomain(*error, NSURLErrorCannotDecodeContentData, AFURLResponseSerializationErrorDomain)) {
- return nil;
- }
- }
- NSError *serializationError = nil;
- NSXMLDocument *document = [[NSXMLDocument alloc] initWithData:data options:self.options error:&serializationError];
- if (!document)
- {
- if (error) {
- *error = AFErrorWithUnderlyingError(serializationError, *error);
- }
- return nil;
- }
-
- return document;
- }
- #pragma mark - NSSecureCoding
- - (instancetype)initWithCoder:(NSCoder *)decoder {
- self = [super initWithCoder:decoder];
- if (!self) {
- return nil;
- }
- self.options = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(options))] unsignedIntegerValue];
- return self;
- }
- - (void)encodeWithCoder:(NSCoder *)coder {
- [super encodeWithCoder:coder];
- [coder encodeObject:@(self.options) forKey:NSStringFromSelector(@selector(options))];
- }
- #pragma mark - NSCopying
- - (instancetype)copyWithZone:(NSZone *)zone {
- AFXMLDocumentResponseSerializer *serializer = [super copyWithZone:zone];
- serializer.options = self.options;
- return serializer;
- }
- @end
- #endif
- #pragma mark -
- @implementation AFPropertyListResponseSerializer
- + (instancetype)serializer {
- return [self serializerWithFormat:NSPropertyListXMLFormat_v1_0 readOptions:0];
- }
- + (instancetype)serializerWithFormat:(NSPropertyListFormat)format
- readOptions:(NSPropertyListReadOptions)readOptions
- {
- AFPropertyListResponseSerializer *serializer = [[self alloc] init];
- serializer.format = format;
- serializer.readOptions = readOptions;
- return serializer;
- }
- - (instancetype)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- self.acceptableContentTypes = [[NSSet alloc] initWithObjects:@"application/x-plist", nil];
- return self;
- }
- #pragma mark - AFURLResponseSerialization
- - (id)responseObjectForResponse:(NSURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- if (![self validateResponse:(NSHTTPURLResponse *)response data:data error:error]) {
- if (!error || AFErrorOrUnderlyingErrorHasCodeInDomain(*error, NSURLErrorCannotDecodeContentData, AFURLResponseSerializationErrorDomain)) {
- return nil;
- }
- }
- if (!data) {
- return nil;
- }
-
- NSError *serializationError = nil;
-
- id responseObject = [NSPropertyListSerialization propertyListWithData:data options:self.readOptions format:NULL error:&serializationError];
-
- if (!responseObject)
- {
- if (error) {
- *error = AFErrorWithUnderlyingError(serializationError, *error);
- }
- return nil;
- }
- return responseObject;
- }
- #pragma mark - NSSecureCoding
- + (BOOL)supportsSecureCoding {
- return YES;
- }
- - (instancetype)initWithCoder:(NSCoder *)decoder {
- self = [super initWithCoder:decoder];
- if (!self) {
- return nil;
- }
- self.format = (NSPropertyListFormat)[[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(format))] unsignedIntegerValue];
- self.readOptions = [[decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(readOptions))] unsignedIntegerValue];
- return self;
- }
- - (void)encodeWithCoder:(NSCoder *)coder {
- [super encodeWithCoder:coder];
- [coder encodeObject:@(self.format) forKey:NSStringFromSelector(@selector(format))];
- [coder encodeObject:@(self.readOptions) forKey:NSStringFromSelector(@selector(readOptions))];
- }
- #pragma mark - NSCopying
- - (instancetype)copyWithZone:(NSZone *)zone {
- AFPropertyListResponseSerializer *serializer = [super copyWithZone:zone];
- serializer.format = self.format;
- serializer.readOptions = self.readOptions;
- return serializer;
- }
- @end
- #pragma mark -
- #if TARGET_OS_IOS || TARGET_OS_TV || TARGET_OS_WATCH
- #import <CoreGraphics/CoreGraphics.h>
- #import <UIKit/UIKit.h>
- @interface UIImage (AFNetworkingSafeImageLoading)
- + (UIImage *)af_safeImageWithData:(NSData *)data;
- @end
- static NSLock* imageLock = nil;
- @implementation UIImage (AFNetworkingSafeImageLoading)
- + (UIImage *)af_safeImageWithData:(NSData *)data {
- UIImage* image = nil;
- static dispatch_once_t onceToken;
- dispatch_once(&onceToken, ^{
- imageLock = [[NSLock alloc] init];
- });
-
- [imageLock lock];
- image = [UIImage imageWithData:data];
- [imageLock unlock];
- return image;
- }
- @end
- static UIImage * AFImageWithDataAtScale(NSData *data, CGFloat scale) {
- UIImage *image = [UIImage af_safeImageWithData:data];
- if (image.images) {
- return image;
- }
-
- return [[UIImage alloc] initWithCGImage:[image CGImage] scale:scale orientation:image.imageOrientation];
- }
- static UIImage * AFInflatedImageFromResponseWithDataAtScale(NSHTTPURLResponse *response, NSData *data, CGFloat scale) {
- if (!data || [data length] == 0) {
- return nil;
- }
- CGImageRef imageRef = NULL;
- CGDataProviderRef dataProvider = CGDataProviderCreateWithCFData((__bridge CFDataRef)data);
- if ([response.MIMEType isEqualToString:@"image/png"]) {
- imageRef = CGImageCreateWithPNGDataProvider(dataProvider, NULL, true, kCGRenderingIntentDefault);
- } else if ([response.MIMEType isEqualToString:@"image/jpeg"]) {
- imageRef = CGImageCreateWithJPEGDataProvider(dataProvider, NULL, true, kCGRenderingIntentDefault);
- if (imageRef) {
- CGColorSpaceRef imageColorSpace = CGImageGetColorSpace(imageRef);
- CGColorSpaceModel imageColorSpaceModel = CGColorSpaceGetModel(imageColorSpace);
- // CGImageCreateWithJPEGDataProvider does not properly handle CMKY, so fall back to AFImageWithDataAtScale
- if (imageColorSpaceModel == kCGColorSpaceModelCMYK) {
- CGImageRelease(imageRef);
- imageRef = NULL;
- }
- }
- }
- CGDataProviderRelease(dataProvider);
- UIImage *image = AFImageWithDataAtScale(data, scale);
- if (!imageRef) {
- if (image.images || !image) {
- return image;
- }
- imageRef = CGImageCreateCopy([image CGImage]);
- if (!imageRef) {
- return nil;
- }
- }
- size_t width = CGImageGetWidth(imageRef);
- size_t height = CGImageGetHeight(imageRef);
- size_t bitsPerComponent = CGImageGetBitsPerComponent(imageRef);
- if (width * height > 1024 * 1024 || bitsPerComponent > 8) {
- CGImageRelease(imageRef);
- return image;
- }
- // CGImageGetBytesPerRow() calculates incorrectly in iOS 5.0, so defer to CGBitmapContextCreate
- size_t bytesPerRow = 0;
- CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
- CGColorSpaceModel colorSpaceModel = CGColorSpaceGetModel(colorSpace);
- CGBitmapInfo bitmapInfo = CGImageGetBitmapInfo(imageRef);
- if (colorSpaceModel == kCGColorSpaceModelRGB) {
- uint32_t alpha = (bitmapInfo & kCGBitmapAlphaInfoMask);
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wassign-enum"
- if (alpha == kCGImageAlphaNone) {
- bitmapInfo &= ~kCGBitmapAlphaInfoMask;
- bitmapInfo |= kCGImageAlphaNoneSkipFirst;
- } else if (!(alpha == kCGImageAlphaNoneSkipFirst || alpha == kCGImageAlphaNoneSkipLast)) {
- bitmapInfo &= ~kCGBitmapAlphaInfoMask;
- bitmapInfo |= kCGImageAlphaPremultipliedFirst;
- }
- #pragma clang diagnostic pop
- }
- CGContextRef context = CGBitmapContextCreate(NULL, width, height, bitsPerComponent, bytesPerRow, colorSpace, bitmapInfo);
- CGColorSpaceRelease(colorSpace);
- if (!context) {
- CGImageRelease(imageRef);
- return image;
- }
- CGContextDrawImage(context, CGRectMake(0.0f, 0.0f, width, height), imageRef);
- CGImageRef inflatedImageRef = CGBitmapContextCreateImage(context);
- CGContextRelease(context);
- UIImage *inflatedImage = [[UIImage alloc] initWithCGImage:inflatedImageRef scale:scale orientation:image.imageOrientation];
- CGImageRelease(inflatedImageRef);
- CGImageRelease(imageRef);
- return inflatedImage;
- }
- #endif
- @implementation AFImageResponseSerializer
- - (instancetype)init {
- self = [super init];
- if (!self) {
- return nil;
- }
- self.acceptableContentTypes = [[NSSet alloc] initWithObjects:@"image/tiff", @"image/jpeg", @"image/gif", @"image/png", @"image/ico", @"image/x-icon", @"image/bmp", @"image/x-bmp", @"image/x-xbitmap", @"image/x-win-bitmap", nil];
- #if TARGET_OS_IOS || TARGET_OS_TV
- self.imageScale = [[UIScreen mainScreen] scale];
- self.automaticallyInflatesResponseImage = YES;
- #elif TARGET_OS_WATCH
- self.imageScale = [[WKInterfaceDevice currentDevice] screenScale];
- self.automaticallyInflatesResponseImage = YES;
- #endif
- return self;
- }
- #pragma mark - AFURLResponseSerializer
- - (id)responseObjectForResponse:(NSURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- if (![self validateResponse:(NSHTTPURLResponse *)response data:data error:error]) {
- if (!error || AFErrorOrUnderlyingErrorHasCodeInDomain(*error, NSURLErrorCannotDecodeContentData, AFURLResponseSerializationErrorDomain)) {
- return nil;
- }
- }
- #if TARGET_OS_IOS || TARGET_OS_TV || TARGET_OS_WATCH
- if (self.automaticallyInflatesResponseImage) {
- return AFInflatedImageFromResponseWithDataAtScale((NSHTTPURLResponse *)response, data, self.imageScale);
- } else {
- return AFImageWithDataAtScale(data, self.imageScale);
- }
- #else
- // Ensure that the image is set to it's correct pixel width and height
- NSBitmapImageRep *bitimage = [[NSBitmapImageRep alloc] initWithData:data];
- NSImage *image = [[NSImage alloc] initWithSize:NSMakeSize([bitimage pixelsWide], [bitimage pixelsHigh])];
- [image addRepresentation:bitimage];
- return image;
- #endif
- return nil;
- }
- #pragma mark - NSSecureCoding
- + (BOOL)supportsSecureCoding {
- return YES;
- }
- - (instancetype)initWithCoder:(NSCoder *)decoder {
- self = [super initWithCoder:decoder];
- if (!self) {
- return nil;
- }
- #if TARGET_OS_IOS || TARGET_OS_TV || TARGET_OS_WATCH
- NSNumber *imageScale = [decoder decodeObjectOfClass:[NSNumber class] forKey:NSStringFromSelector(@selector(imageScale))];
- #if CGFLOAT_IS_DOUBLE
- self.imageScale = [imageScale doubleValue];
- #else
- self.imageScale = [imageScale floatValue];
- #endif
- self.automaticallyInflatesResponseImage = [decoder decodeBoolForKey:NSStringFromSelector(@selector(automaticallyInflatesResponseImage))];
- #endif
- return self;
- }
- - (void)encodeWithCoder:(NSCoder *)coder {
- [super encodeWithCoder:coder];
- #if TARGET_OS_IOS || TARGET_OS_TV || TARGET_OS_WATCH
- [coder encodeObject:@(self.imageScale) forKey:NSStringFromSelector(@selector(imageScale))];
- [coder encodeBool:self.automaticallyInflatesResponseImage forKey:NSStringFromSelector(@selector(automaticallyInflatesResponseImage))];
- #endif
- }
- #pragma mark - NSCopying
- - (instancetype)copyWithZone:(NSZone *)zone {
- AFImageResponseSerializer *serializer = [super copyWithZone:zone];
- #if TARGET_OS_IOS || TARGET_OS_TV || TARGET_OS_WATCH
- serializer.imageScale = self.imageScale;
- serializer.automaticallyInflatesResponseImage = self.automaticallyInflatesResponseImage;
- #endif
- return serializer;
- }
- @end
- #pragma mark -
- @interface AFCompoundResponseSerializer ()
- @property (readwrite, nonatomic, copy) NSArray *responseSerializers;
- @end
- @implementation AFCompoundResponseSerializer
- + (instancetype)compoundSerializerWithResponseSerializers:(NSArray *)responseSerializers {
- AFCompoundResponseSerializer *serializer = [[self alloc] init];
- serializer.responseSerializers = responseSerializers;
- return serializer;
- }
- #pragma mark - AFURLResponseSerialization
- - (id)responseObjectForResponse:(NSURLResponse *)response
- data:(NSData *)data
- error:(NSError *__autoreleasing *)error
- {
- for (id <AFURLResponseSerialization> serializer in self.responseSerializers) {
- if (![serializer isKindOfClass:[AFHTTPResponseSerializer class]]) {
- continue;
- }
- NSError *serializerError = nil;
- id responseObject = [serializer responseObjectForResponse:response data:data error:&serializerError];
- if (responseObject) {
- if (error) {
- *error = AFErrorWithUnderlyingError(serializerError, *error);
- }
- return responseObject;
- }
- }
- return [super responseObjectForResponse:response data:data error:error];
- }
- #pragma mark - NSSecureCoding
- + (BOOL)supportsSecureCoding {
- return YES;
- }
- - (instancetype)initWithCoder:(NSCoder *)decoder {
- self = [super initWithCoder:decoder];
- if (!self) {
- return nil;
- }
- NSSet *classes = [NSSet setWithArray:@[[NSArray class], [AFHTTPResponseSerializer <AFURLResponseSerialization> class]]];
- self.responseSerializers = [decoder decodeObjectOfClasses:classes forKey:NSStringFromSelector(@selector(responseSerializers))];
- return self;
- }
- - (void)encodeWithCoder:(NSCoder *)coder {
- [super encodeWithCoder:coder];
- [coder encodeObject:self.responseSerializers forKey:NSStringFromSelector(@selector(responseSerializers))];
- }
- #pragma mark - NSCopying
- - (instancetype)copyWithZone:(NSZone *)zone {
- AFCompoundResponseSerializer *serializer = [super copyWithZone:zone];
- serializer.responseSerializers = self.responseSerializers;
- return serializer;
- }
- @end
|