ios - 带有 NSURLSession 的 HTTP 基本认证

Question

我正在尝试使用 NSURLSession 实现 HTTP Basic Auth，但我遇到了几个问题。请在回答之前阅读整个问题，我怀疑这是与其他问题的重复。

根据我运行的测试，NSURLSession 的行为如下:

• 第一个请求总是在没有 Authorization header 的情况下发出。
• 如果第一个请求因 401 Unauthorized 响应和 WWW-Authenticate Basic realm=... header 而失败，则会自动重试。
• 在重试请求之前， session 将尝试通过查看 session 配置的 NSURLCredentialStorage 或调用 URLSession:task:didReceiveChallenge:completionHandler: 来获取凭证委托(delegate)方法(或两者)。
• 如果可以获取凭据，则使用正确的 Authorization header 重试请求。如果不是，则在没有 header 的情况下重试(这很奇怪，因为在这种情况下，这是完全相同的请求)。
• 如果第二个请求成功，任务会透明地报告为成功，您甚至不会收到请求已尝试两次的通知。如果不是，则报告第二个请求失败(但不是第一个)。

我对这种行为的问题是我通过多部分请求将大文件上传到我的服务器，所以当请求被尝试两次时，整个 POST 主体被发送两次，这是一个可怕的开销。

我已尝试将 Authorization header 手动添加到 session 配置的 httpAdditionalHeaders 中，但只有在 之前 session 已创建。之后尝试修改 session.configuration.httpAdditionalHeaders 是行不通的。此外，文档明确指出不应手动设置 Authorization header 。

---

所以我的问题是:如果我需要在获得凭据之前开始 session ，并且如果我想确保请求始终是通过正确的授权发出的header 第一次出现，怎么办？

---

这是我用于测试的代码示例。您可以用它重现我上面描述的所有行为。

请注意，为了能够看到双重请求，您需要使用自己的 http 服务器并记录请求，或者通过记录所有请求的代理进行连接(为此我使用了 Charles Proxy)

class URLSessionTest: NSObject, URLSessionDelegate
{
    static let shared = URLSessionTest()

    func start()
    {
        let requestURL = URL(string: "https://httpbin.org/basic-auth/username/password")!
        let credential = URLCredential(user: "username", password: "password", persistence: .forSession)
        let protectionSpace = URLProtectionSpace(host: "httpbin.org", port: 443, protocol: NSURLProtectionSpaceHTTPS, realm: "Fake Realm", authenticationMethod: NSURLAuthenticationMethodHTTPBasic)

        let useHTTPHeader = false
        let useCredentials = true
        let useCustomCredentialsStorage = false
        let useDelegateMethods = true

        let sessionConfiguration = URLSessionConfiguration.default

        if (useHTTPHeader) {
            let authData = "\(credential.user!):\(credential.password!)".data(using: .utf8)!
            let authValue = "Basic " + authData.base64EncodedString()
            sessionConfiguration.httpAdditionalHeaders = ["Authorization": authValue]
        }
        if (useCredentials) {
            if (useCustomCredentialsStorage) {
                let urlCredentialStorage = URLCredentialStorage()
                urlCredentialStorage.set(credential, for: protectionSpace)
                sessionConfiguration.urlCredentialStorage = urlCredentialStorage
            } else {
                sessionConfiguration.urlCredentialStorage?.set(credential, for: protectionSpace)
            }
        }

        let delegate = useDelegateMethods ? self : nil
        let session = URLSession(configuration: sessionConfiguration, delegate: delegate, delegateQueue: nil)

        self.makeBasicAuthTest(url: requestURL, session: session) {
            self.makeBasicAuthTest(url: requestURL, session: session) {
                DispatchQueue.main.asyncAfter(deadline: .now() + 61.0) {
                    self.makeBasicAuthTest(url: requestURL, session: session) {}
                }
            }
        }
    }

    func makeBasicAuthTest(url: URL, session: URLSession, completion: @escaping () -> Void)
    {
        let task = session.dataTask(with: url) { (data, response, error) in
            if let response = response {
                print("response : \(response)")
            }
            if let data = data {
                if let json = try? JSONSerialization.jsonObject(with: data, options: .allowFragments) {
                    print("json : \(json)")
                } else if data.count > 0, let string = String(data: data, encoding: .utf8) {
                    print("string : \(string)")
                } else {
                    print("data : \(data)")
                }
            }
            if let error = error {
                print("error : \(error)")
            }
            print()
            DispatchQueue.main.async(execute: completion)
        }
        task.resume()
    }

    @objc(URLSession:didReceiveChallenge:completionHandler:)
    func urlSession(_ session: URLSession, didReceive challenge: URLAuthenticationChallenge, completionHandler: @escaping (URLSession.AuthChallengeDisposition, URLCredential?) -> Swift.Void)
    {
        print("Session authenticationMethod: \(challenge.protectionSpace.authenticationMethod)")
        if (challenge.protectionSpace.authenticationMethod == NSURLAuthenticationMethodHTTPBasic) {
            let credential = URLCredential(user: "username", password: "password", persistence: .forSession)
            completionHandler(.useCredential, credential)
        } else {
            completionHandler(.performDefaultHandling, nil)
        }
    }

    @objc(URLSession:task:didReceiveChallenge:completionHandler:)
    func urlSession(_ session: URLSession, task: URLSessionTask, didReceive challenge: URLAuthenticationChallenge, completionHandler: @escaping (URLSession.AuthChallengeDisposition, URLCredential?) -> Swift.Void)
    {
        print("Task authenticationMethod: \(challenge.protectionSpace.authenticationMethod)")
        if (challenge.protectionSpace.authenticationMethod == NSURLAuthenticationMethodHTTPBasic) {
            let credential = URLCredential(user: "username", password: "password", persistence: .forSession)
            completionHandler(.useCredential, credential)
        } else {
            completionHandler(.performDefaultHandling, nil)
        }
    }
}

注意 1:当连续向同一个端点发出多个请求时，我上面描述的行为仅涉及第一个请求。第一次使用正确的 Authorization header 尝试后续请求。但是，如果您等待一段时间(大约 1 分钟)， session 将返回默认行为(第一次请求尝试两次)。

注意 2:这没有直接关系，但是为 session 配置的 urlCredentialStorage 使用自定义 NSURLCredentialStorage 似乎没有工作。仅使用默认值(根据文档，这是共享的 NSURLCredentialStorage)有效。

注意 3:我试过使用 Alamofire，但由于它基于 NSURLSession，所以它的行为方式完全相同。

Answer 1

如果可能，服务器应该在客户端完成发送主体之前很久就以错误响应。然而，在许多高级服务器端语言中，这很困难，而且即使您这样做也不能保证上传会停止。

真正的问题是您正在使用单个 POST 请求执行大型上传。这使得身份验证成为问题，并且如果连接在上传中途断开，也会阻止任何有用的继续上传。分块上传基本上可以解决您的所有问题:

• 对于您的第一个请求，只发送适合的数量而不添加额外的以太网数据包，即计算您的典型 header 大小，模数 1500 字节，添加几十个字节以进行良好测量，从 1500 中减去，并为您的第一个 block 硬编码该大小。最多，您浪费了几个数据包。

• 对于后续 block ，将大小调大。

• 当请求失败时，询问服务器它得到了多少，然后从上传停止的地方重试。

• 上传完成后发出请求通知服务器。

• 使用 cron 作业或其他方式定期清除服务器端的部分上传。

也就是说，如果您无法控制服务器端，通常的解决方法是在您的 POST 请求之前发送经过身份验证的 GET 请求。这最大限度地减少了浪费的数据包，同时只要网络可靠，大部分时间仍然可以正常工作。