思考NSURLProtocol无法拦截WKWebview请求的解决思路

1.hook XMLHTTPRequest

原理简单说就是hook XMLHTTPRequest的send方法和open函数，然后在里面重组url和body参数，然后重新发起请求，这边有个难点，就是send和open是异步的，所以需要等hook到open中的参数后再发起请求。

2.WKURLSchemeHandler

WKURLSchemeHandler是iOS11就推出的，用于处理自定义请求的方案，不过并不能处理HTTP、HTTPS等常规scheme。但是现在有另外一个方法，就是去 WKWebview的initWithFrame:方法，然后设置一个WKURLSchemeHandler给WKWebview实例，这样就能截获WKWebview的HTTP和HTTPS请求了。范例如下

//
//  ViewController.m
//  WKWebviewURLSchemeHandlerProject
//
//  Created by jerrold on 2019/3/5.
//  Copyright © 2019 jerrold. All rights reserved.
//

#import "ViewController.h"
#import 

@interface CustomURLSchemeHandler : NSObject
@end

@implementation CustomURLSchemeHandler
//这里拦截到URLScheme为customScheme的请求后，读取本地图片test.jpg，并返回给WKWebView显示
- (void)webView:(WKWebView *)webView startURLSchemeTask:(id)urlSchemeTask{
    NSURLRequest *request = urlSchemeTask.request;
    //在下面处理拦截到的HTTP和HTTPS请求,处理完后再次发起请求, 注意要处理didReceiveResponse:  didReceiveData: didFinish三个方法回调
    if ([request.URL.scheme containsString:@"https"]) {
        
    }else if ([request.URL.scheme containsString:@"http"]){
        
    }
//    UIImage *image = [UIImage imageNamed:@"test.jpg"];
//    NSData *data = UIImageJPEGRepresentation(image, 1.0);
//    NSURLResponse *response = [[NSURLResponse alloc] initWithURL:request.URL MIMEType:@"image/jpeg" expectedContentLength:data.length textEncodingName:nil];
//    [urlSchemeTask didReceiveResponse:response];
//    [urlSchemeTask didReceiveData:data];
//    [urlSchemeTask didFinish];
}


- (void)webView:(WKWebView *)webVie stopURLSchemeTask:(id)urlSchemeTask {
}
@end
@interface ViewController ()

@end

@implementation ViewController
- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
    //如下是以APP为视角写的处理WKWebview请求的方式,如果要AOP,就需要hook WKWebview的initwithFrame方法,在里面处理WKWebview的请求
    WKWebViewConfiguration *configuration = [WKWebViewConfiguration new];
    CustomURLSchemeHandler *handler = [[CustomURLSchemeHandler alloc]init];
    NSMutableDictionary *handlers = [configuration valueForKey:@"_urlSchemeHandlers"];
    handlers[@"https"] = handler;//修改handler,将HTTP和HTTPS也一起拦截
    handlers[@"http"] = handler;
    WKWebView *webView = [[WKWebView alloc] initWithFrame:self.view.bounds configuration:configuration];
    self.view = webView;
    [webView loadRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:@"http://www.test.com"]]];
}
@end
复制代码

不过，这种方案和第一种有共性，就是利用了私有API，很可能导致被ban，毕竟apple最近审查非常严苛。但是起码比第一种方案好，不用担心body大小的问题，不用担心影响浏览器性能。

参考文献

1.WebKitでPrivateAPI使ってProxyを実装する

2.Intercepting UIWebView & WKWebView — Custom URL Scheme handling

3.iOS 11 WKWebView新特性

转载于:https://juejin.im/post/5cac5046518825710d76c4e1