Merge pull request #2 from readdle/fix/call-auth-challenge-if-delegate-is-presented

Do not use AuthChallenge if no task delegate presented

Author: andriydruk

Foundation/URLSession/URLSessionTask.swift

@@ -659,7 +659,8 @@ extension _ProtocolClient: URLProtocolClient {
         guard let session = task.session as? URLSession else { fatalError("session cannot be nil") }
         guard let response = task.response as? HTTPURLResponse else { fatalError("No response") }
 
-        if response.statusCode == 401 {
+        
+        if response.statusCode == 401, `protocol`.containsTaskDelegate() {
             // Concat protection space from header with all possibles protection spaces
             if !task.protectionSpacesInited { // init protection spaces
                 var allPossibleProtectionSpaces = AuthProtectionSpace.createAllPossible(using: response)
@@ -683,6 +684,7 @@ extension _ProtocolClient: URLProtocolClient {
                 return
             }
         }
+    
 
         switch session.behaviour(for: task) {
         case .taskDelegate(let delegate):
@@ -746,7 +748,7 @@ extension _ProtocolClient: URLProtocolClient {
         }
         let certificateErrors = [NSURLErrorServerCertificateUntrusted, NSURLErrorServerCertificateWrongHost]
 
-        if certificateErrors.contains(error._code) {
+        if certificateErrors.contains(error._code) && `protocol`.containsTaskDelegate() {
             let protectionSpace = URLProtectionSpace(host: "",
                     port: 443,
                     protocol: "https",
@@ -763,10 +765,9 @@ extension _ProtocolClient: URLProtocolClient {
 
                 task.previousFailureCount += 1
                 urlProtocol(`protocol`, didReceive: authenticationChallenge)
+                return
             }
-            return
         }
-
         urlProtocol(task: task, didFailWithError: error)
     }
 
@@ -838,6 +839,21 @@ private extension URLSessionTask {
     }
 }
 
+fileprivate extension URLProtocol {
+    func containsTaskDelegate() -> Bool {
+        guard let task = self.task else { return false }
+        guard let session = task.session as? URLSession else { return false }
+
+        switch session.behaviour(for: task) {
+        case .taskDelegate(_):
+            return true
+            
+        default:
+            return false
+        }
+    }
+}
+
 extension URLProtocol {
     enum _PropertyKey: String {
         case responseData

Foundation/URLSession/http/HTTPBodySource.swift

@@ -126,6 +126,7 @@ extension _HTTPBodyStreamSource: _HTTPBodySource {
         if inputStream.hasBytesAvailable {
             let buffer = UnsafeMutableRawBufferPointer.allocate(count: length)
             guard let pointer = buffer.baseAddress?.assumingMemoryBound(to: UInt8.self) else {
+                buffer.deallocate()
                 return .error
             }
             let readBytes = self.inputStream.read(pointer, maxLength: length)