将NSURLProtocol与NSURLSession一起使用

Question

我的应用程序使用NSURLConnection与服务器通信。我们使用https进行通信。为了在一个地方处理来自所有请求的身份验证，我使用NSURLProtocol并在该类的委托中处理身份验证。现在我决定使用NSURLSession而不是NSURLConnection。我正在尝试让NSURLProtocol与NSURLSession合作我创建了一个任务并使用了NSURLProtocol

NSMutableURLRequest *sampleRequest = [[NSMutableURLRequest alloc]initWithURL:someURL];
NSURLSessionConfiguration *configuration = [NSURLSessionConfiguration defaultSessionConfiguration];
configuration.protocolClasses = @[[CustomHTTPSProtocol class]];
NSURLSession *session = [NSURLSession sessionWithConfiguration:configuration delegate:self delegateQueue:nil];
NSURLSessionDataTask *task = [session dataTaskWithRequest:checkInInfoRequest];
[task resume];

CustomHTTPSProtocol这是我的NSURLProtocol类看起来像这样

static NSString * const CustomHTTPSProtocolHandledKey = @"CustomHTTPSProtocolHandledKey";

@interface CustomHTTPSProtocol () <NSURLSessionDataDelegate,NSURLSessionTaskDelegate,NSURLSessionDelegate>

@property (nonatomic, strong) NSURLSessionDataTask *connection;
@property (nonatomic, strong) NSMutableData *mutableData;

@end

@implementation CustomHTTPSProtocol

+ (BOOL)canInitWithRequest:(NSURLRequest *)request {
    if ([NSURLProtocol propertyForKey:CustomHTTPSProtocolHandledKey inRequest:request]) {
        return NO;
    }
    return [[[[request URL]scheme]lowercaseString]isEqualToString:@"https"];
}

+ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {
    return request;
}

- (void) startLoading {
    NSMutableURLRequest *newRequest = [self.request mutableCopy];
    [NSURLProtocol setProperty:@YES forKey:CustomHTTPSProtocolHandledKey inRequest:newRequest];

    NSURLSession*session = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration] delegate:self delegateQueue:nil];
    self.connection = [session dataTaskWithRequest:newRequest];
    [self.connection resume];
    self.mutableData = [[NSMutableData alloc] init];
}

- (void) stopLoading {
    [self.connection cancel];
    self.mutableData = nil;
}

-(void)URLSession:(NSURLSession *)session didReceiveChallenge:(NSURLAuthenticationChallenge *)challenge completionHandler:(void (^)(NSURLSessionAuthChallengeDisposition disposition, NSURLCredential *credential))completionHandler {
    NSLog(@"challenge..%@",challenge.protectionSpace.authenticationMethod);
    if ([challenge.protectionSpace.authenticationMethod isEqualToString:NSURLAuthenticationMethodServerTrust]) {
        [challenge.sender useCredential:[NSURLCredential credentialForTrust:challenge.protectionSpace.serverTrust] forAuthenticationChallenge:challenge];
    }
    else {
        [challenge.sender continueWithoutCredentialForAuthenticationChallenge:challenge];
    }
}

- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveData:(NSData *)data {
    [self.client URLProtocol:self didLoadData:data];
    NSLog(@"data ...%@  ",data); //handle data here
    [self.mutableData appendData:data];
}

- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error {
    if (!error) {
        [self.client URLProtocolDidFinishLoading:self];
    }
    else {
        NSLog(@"error ...%@  ",error);
        [self.client URLProtocol:self didFailWithError:error];
    }
}

@end

调用开始加载并且还完成了身份验证质询，但在此之后立即调用停止加载。

一段时间后返回错误代码-999“已取消”。 didReceiveData未被调用。

Note:NSURLProtocol and the Authentication Process worked fine with NSURLConnection.

我错过了什么？我的问题是

1. 注册[NSURLProtocol registerClass：[CustomHTTPSProtocol class]];使用NSURLConnection工作正常，但如何使用NSURLSession全局抵御NSURLProtocol？
2. 为什么NSURLProtocol中的请求失败（使用URL链接和逻辑使用URL连接）以及如何使NSURLProtocol使用URLSession？

请帮助我，如果您想了解更多细节，请告诉我。

Answer 1

我知道这是旧的，但你遇到的问题是你的didReceiveChallenge方法。您正在通过调用结束该方法

[challenge.sender continueWithoutCredentialForAuthenticationChallenge:challenge];

要么

[challenge.sender useCredential:[NSURLCredential credentialForTrust:challenge.protectionSpace.serverTrust] forAuthenticationChallenge:challenge];

您需要做的是使用完成处理程序来发送结果。它看起来像这样：

completionHandler(.UseCredential, NSURLCredential(forTrust: challenge.protectionSpace.serverTrust)

要么

completionHandler(.PerformDefaultHandling, nil)

这些都在Swift 2.0中，但应该很好地转换为Obj-C。

Answer 2

使用NSUrlsession注册自定义NSURLProtocol与使用NSURLConnection的方式相同。

与自定义NSUrlProtocol一起使用时，NSURLSession Api（NS​​URLSessionDataTask和其他任务类）无法正确处理HTTP身份验证挑战。这在iOS 7.x上按预期工作，在iOS 8.x中被破坏。发现了一个苹果雷达，我的雷达被关闭，说它是另一个雷达的副本，我仍然看到原始雷达仍然打开。所以我认为这个问题目前尚未被苹果公司解决。

希望我的答案还不算太晚:)

Answer 3

我模糊的回忆是NSURLSession自动使用任何全球注册的协议。所以你不需要再次注册，但这样做也不应该受到伤害。

我想知道URL请求是否被复制，在这种情况下，您的自定义标记可能不起作用，并且您可能会看到无限递归，直到它达到某个内部限制。您是否尝试过设置请求标头？