iOS 通过NSURLProtocol拦截WKWebView网络请求

以前NSURLProtocol可以���接拦截UIWebView，后面升级成WKWebView发现拦截不到了

有细心爱研究的老铁发现了 [WKBrowsingContextController  registerSchemeForCustomProtocol:]  这个函数的存在

所以还是可以拦截的

直接上步骤

1.在控制器或者你喜欢的地方注册NSURLProtocol(调用以下代码)

    Class cls = NSClassFromString(@"WKBrowsingContextController");    SEL sel = NSSelectorFromString(@"registerSchemeForCustomProtocol:");    if ([(id)cls respondsToSelector:sel]) {        [(id)cls performSelector:sel withObject:@"http"];        [(id)cls performSelector:sel withObject:@"https"];    }    [NSURLProtocol registerClass:[CustomProtocol class]];

2.在你自定义的NSURLProtocol类中实现拦截

+ (BOOL)canInitWithRequest:(NSURLRequest *)request{        NSString *abstring = request.URL.absoluteString;        if ([abstring containsString:@"你喜欢的标记"]) {        return YES;    }        return NO;}+ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request{    return request;}

3.在你自定义的NSURLProtocol类中拦截后替换请求的内容，以下以图片为例

- (void)startLoading{        NSString *urlPathString = super.request.URL.absoluteString;        if ([urlPathString containsString:@"你喜欢的标记"]) {                urlPathString = [urlPathString stringByReplacingOccurrencesOfString:@"你喜欢的标记" withString:@"你喜欢的标记2"];    }            NSData *imageData = [NSData dataWithContentsOfURL:[NSURL URLWithString:urlPathString]];        NSMutableDictionary *header = [NSMutableDictionary dictionary];    header[@"Content-Length"] = [NSString stringWithFormat:@"%ld",imageData.length];        NSHTTPURLResponse *response = [[NSHTTPURLResponse alloc] initWithURL:self.request.URL               statusCode:200 HTTPVersion:@"1.1" headerFields:header];    //回调    [self.client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageAllowed];    [self.client URLProtocol:self didLoadData:imageData];    [self.client URLProtocolDidFinishLoading:self];}- (void)stopLoading{    NSLog(@"stop");}