Merge pull request #5 from emsquared/master

Add sanity check for _response in ImageSizeCheckClient.m
psychs · Jan 23, 2016 · b7dc5e8 · b7dc5e8
2 parents 4a9d249 + 18aef57
commit b7dc5e8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Classes/Library/ImageSizeCheckClient.m b/Classes/Library/ImageSizeCheckClient.m
@@ -66,6 +66,13 @@ - (void)connectionDidFinishLoading:(NSURLConnection *)sender
 {
     if (_conn != sender) return;
 
+	/* If a user clicks a link that redirects to a data image, then 
+	 this value will not be NSHTTPURLResponse which results in a crash
+	 when trying to make a call to -allHeaderFields */
+	if ([_response isKindOfClass:[NSHTTPURLResponse class]] == NO) {
+		return;
+	}
+
     long long contentLength = 0;
     NSString* contentType;
     int statusCode = [_response statusCode];