2

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

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

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

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

我试图手动将Authorization标头添加到httpAdditionalHeaders会话配置中,但只有在创建会话之前设置了属性时它才有效。之后尝试修改session.configuration.httpAdditionalHeaders是行不通的。此外,文档清楚地表明Authorization不应手动设置标题。


所以我的问题是:如果我需要在获得凭据之前开始会话,并且如果我想确保 Authorization第一次总是使用正确的标头发出请求,我该怎么办?


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

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

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第一次尝试使用正确的标头尝试后续请求。但是,如果您等待一段时间(大约 1 分钟),会话将返回默认行为(第一次请求尝试了两次)。

注意 2:这没有直接关系,但是使用自定义NSURLCredentialStorageurlCredentialStorage会话配置似乎不起作用。仅使用默认值(NSURLCredentialStorage根据文档共享)有效。

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

4

1 回答 1

2

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

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

  • 对于您的第一个请求,仅发送适合的数量而不添加额外的以太网数据包,即计算您的典型标头大小,以 1500 字节为模,添加几十个字节以获得良好的度量,从 1500 中减去,然后硬编码该大小对于你的第一块。最多,您浪费了几个数据包。

  • 对于后续的块,将大小调大。

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

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

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

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

于 2017-03-20T23:04:30.280 回答