iOS 通过NSURLProtocol拦截WKWebView网络请求

最新推荐文章于 2024-08-20 08:46:03 发布

chengqiang0414

最新推荐文章于 2024-08-20 08:46:03 发布

阅读量1.5k

点赞数 7

文章标签： WKWebView拦截 WKWebView请求替换

本文链接：https://blog.csdn.net/chengqiang0414/article/details/136323992

版权

文章介绍了如何在WKWebView升级后，通过注册自定义NSURLProtocol并实现拦截功能，包括步骤如在`WKBrowsingContextController`中注册协议、自定义协议的初始化和请求处理，以及如何替换特定请求内容如图片URL。

摘要由CSDN通过智能技术生成

以前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");
}