ios - 使用 NSURLSession 进行 HTTP 基本身份验证

Question

我正在尝试使用 NSURLSession 实现 HTTP 基本身份验证，但遇到了几个问题。请在回答之前阅读整个问题，我怀疑这与其他问题重复。

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

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

我遇到的这种行为的问题是，我通过分段请求将大文件上传到我的服务器，因此当尝试请求两次时，整个 POST 正文会发送两次，这是一个可怕的问题开销。

我尝试将 Authorization header 手动添加到 session 配置的 httpAdditionalHeaders 中，但只有在之前设置该属性时才有效> session 已创建。之后尝试修改 session.configuration.httpAdditionalHeaders 不起作用。此外，文档明确指出不应手动设置 Authorization header 。

---

所以我的问题是:我是否需要在获取凭据之前启动 session ，以及如果我想确保请求始终使用正确的授权第一次标题，我该怎么办？

---

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

请注意，为了能够看到双重请求，您需要使用自己的 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:这没有直接关系，但使用自定义 NSURLCredentialStorage 作为 session 配置的 urlCredentialStorage 似乎并不重要工作。仅使用默认值(根据文档，这是共享的 NSURLCredentialStorage)才有效。

注释 3:我尝试过使用 Alamofire，但由于它基于 NSURLSession，因此其行为方式完全相同。

Answer 1

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

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

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

• 对于后续 block ，增大大小。

• 当请求失败时，询问服务器获得了多少数据，然后从上传中断处重试。

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

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

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