WKWebView的请求拦截和修改实现。技术方法：NSURLProtocol

于 2024-04-08 15:35:04 发布
阅读量578 收藏 1
点赞数 3
文章标签： ios
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u012812881/article/details/137510506
版权
  • 需求，拦截WKWebview中的所有网络请求，并且对亲够Request的httpheader中添加字段token，等信息。
  • 实现技术，利用NSURLProtocol。
  • 首先实现一个继承自NSURLProtocol的自定义类：MYSchemeURLProtocol,完整的代码实现如下：
//类文件

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

#define MYSchemeURLHeaderTokenAdd 0

FOUNDATION_EXTERN NSString *const URLLoadingNotification;

FOUNDATION_EXTERN NSString *const HttpProtocolKey;
FOUNDATION_EXTERN NSString *const HttpsProtocolKey;


@interface MYSchemeURLProtocol : NSURLProtocol
+ (void)registSchemeURLProtocol;
+ (void)unregistSchemeURLProtocol;
@end


@interface NSMutableURLRequest (MYHeaderAppend)
- (BOOL)appendRemoteAccessHeaders;
@end

NS_ASSUME_NONNULL_END

//================.m ===================
#import "MYSchemeURLProtocol.h"
#import "JSON.h"


NSString *const URLLoadingNotification = @"com.zspace.urlLoadingNotification";
static NSString *kURLProtocolHandledKey = @"URLProtocolHandledKey";
NSString *const HttpProtocolKey = @"http";
NSString *const HttpsProtocolKey = @"https";


@interface MYSchemeURLProtocol()<NSURLSessionDelegate>

@property (atomic,strong,readwrite) NSURLSessionDataTask *task;
@property (nonatomic,strong) NSURLSession *session;
@property (nonatomic, strong) NSOperationQueue *queue;

@end

@implementation MYSchemeURLProtocol
+ (void)registSchemeURLProtocol {
    // 防止苹果静态检查 将 WKBrowsingContextController 拆分，然后再拼凑起来
    NSArray *privateStrArr = @[@"Controller", @"Context", @"Browsing", @"K", @"W"];
    NSString *className =  [[[privateStrArr reverseObjectEnumerator] allObjects] componentsJoinedByString:@""];
    Class cls = NSClassFromString(className);
    SEL sel = NSSelectorFromString(@"registerSchemeForCustomProtocol:");
    
    if (cls && sel) {
        if ([(id)cls respondsToSelector:sel]) {
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
            // 注册自定义协议
            // [(id)cls performSelector:sel withObject:@"CustomProtocol"];
            // 注册http协议
            [(id)cls performSelector:sel withObject:HttpProtocolKey];
            // 注册https协议
            //            [(id)cls performSelector:sel withObject:HttpsProtocolKey];
#pragma clang diagnostic pop
        }
    }
    // SechemaURLProtocol 自定义类 继承于 NSURLProtocol
    [NSURLProtocol registerClass:[MYSchemeURLProtocol class]];
}

+ (void)unregistSchemeURLProtocol {
    [NSURLProtocol unregisterClass:[MYSchemeURLProtocol class]];
}

+ (BOOL)canInitWithRequest:(NSURLRequest *)request
{
    NSString *scheme = [[request URL] scheme];
    // 判断是否需要进入自定义加载器
    if ([scheme caseInsensitiveCompare:HttpProtocolKey] == NSOrderedSame
        /*|| [scheme caseInsensitiveCompare:HttpsProtocolKey] == NSOrderedSame*/)
    {
        //看看是否已经处理过了，防止无限循环
        if ([NSURLProtocol propertyForKey:kURLProtocolHandledKey inRequest:request]) {
            NSLogInfo(@"hookRequest-canInitWithRequest NO %@",request.URL);
            return NO;
        }
    }
    NSLogInfo(@"hookRequest-canInitWithRequest YES %@",request.URL);
    return YES;
}

+ (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
    
    NSMutableURLRequest *mutableReqeust = [request mutableCopy];
    // 执行自定义操作，例如添加统一的请求头等
    return mutableReqeust;
}

// 判重
+ (BOOL)requestIsCacheEquivalent:(NSURLRequest *)a toRequest:(NSURLRequest *)b
{
    return [super requestIsCacheEquivalent:a toRequest:b];
}

- (void)startLoading
{
    NSMutableURLRequest *mutableReqeust = [[self request] mutableCopy];
    // 标示改request已经处理过了，防止无限循环
    [NSURLProtocol setProperty:@YES forKey:kURLProtocolHandledKey inRequest:mutableReqeust];
    
#if MYSchemeURLHeaderTokenAdd
    // 处理请求 body
    if (self.request.HTTPBody && [self.request valueForHTTPHeaderField:@"Content-Type"]) {
        [mutableReqeust setHTTPBody:self.request.HTTPBody];
        [mutableReqeust setValue:[self.request valueForHTTPHeaderField:@"Content-Type"] forHTTPHeaderField:@"Content-Type"];
    }
    if([mutableReqeust.URL.absoluteString containsString:@"Users/authenticatebyname"]){
        NSLogInfo(@"跳过");
    }
    NSString *body =  [NTYJSON parse:mutableReqeust.HTTPBody];
    NSLogInfo(@"self.HTTPBody:%@",body);
    NSLogInfo(@"self.HTTPBodyStream:%@",mutableReqeust.HTTPBodyStream);
        
    [mutableReqeust appendRemoteAccessHeaders];
#endif
    
    // 通知更新URL
    [[NSNotificationCenter defaultCenter] postNotificationName:URLLoadingNotification object:mutableReqeust.URL.absoluteString];
    
    NSURLSessionConfiguration *configure = [NSURLSessionConfiguration defaultSessionConfiguration];
    self.session  = [NSURLSession sessionWithConfiguration:configure delegate:self delegateQueue:self.queue];
    self.task = [self.session dataTaskWithRequest:mutableReqeust];
    [self.task resume];
}

- (void)stopLoading
{
    [self.session invalidateAndCancel];
    self.session = nil;
    [[NSNotificationCenter defaultCenter] postNotificationName:URLLoadingNotification object:@""];
    
}

#pragma mark - Getter
- (NSOperationQueue *)queue
{
    if (!_queue) {
        _queue = [[NSOperationQueue alloc] init];
    }
    return _queue;
}
@end

@implementation MYSchemeURLProtocol(NSURLSessionDelegate)

- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error
{
    if (error != nil) {
        NSLogInfo(@"didComplete url:%@",task.originalRequest.URL.absoluteString);
        // 检查是否为 WKWebView 的网络请求
        [self.client URLProtocol:self didFailWithError:error];
    }else
    {
        [self.client URLProtocolDidFinishLoading:self];
    }
}

- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask
didReceiveResponse:(NSURLResponse *)response
 completionHandler:(void (^)(NSURLSessionResponseDisposition disposition))completionHandler
{
    [self.client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
    
    completionHandler(NSURLSessionResponseAllow);
}

- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveData:(NSData *)data
{
    [self.client URLProtocol:self didLoadData:data];
}

- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask willCacheResponse:(NSCachedURLResponse *)proposedResponse completionHandler:(void (^)(NSCachedURLResponse * _Nullable))completionHandler
{
    completionHandler(proposedResponse);
}

//TODO: 重定向
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task willPerformHTTPRedirection:(NSHTTPURLResponse *)response newRequest:(NSURLRequest *)newRequest completionHandler:(void (^)(NSURLRequest *))completionHandler
{
    NSLogInfo(@"hookRequest-willPerformHTTP:%@",newRequest.URL);
    NSMutableURLRequest*    redirectRequest;
    redirectRequest = [newRequest mutableCopy];
    [[self class] removePropertyForKey:kURLProtocolHandledKey inRequest:redirectRequest];
    
#if MYSchemeURLHeaderTokenAdd
    [redirectRequest appendRemoteAccessHeaders];
#endif
    
    [[self client] URLProtocol:self wasRedirectedToRequest:redirectRequest redirectResponse:response];
    
    [self.task cancel];
    [[self client] URLProtocol:self didFailWithError:[NSError errorWithDomain:NSCocoaErrorDomain code:NSUserCancelledError userInfo:nil]];
}


@end




@implementation NSMutableURLRequest (MYHeaderAppend)
/*
 还需要增加两个header：
 Nas-Neigh-Ip
 Nas-Neigh-Port
 分别是目标ip和目标端口，请注意port需要用字符串类型
 */
- (BOOL)appendRemoteAccessHeaders {
    //MARK:MY_token 添加进请求头
    // 添加自定义的 HTTP 头
    NSString *tokenValue = [MYUserService shared].currentUser.token;
    NSString *holdToken = [self.allHTTPHeaderFields valueForKey:@"z-token"];
    if([self.URL.absoluteString containsString:@"Users/authenticatebyname"]){
        NSLogInfo(@"跳过");
    }
    if(isEmpty(holdToken)){
        NSString *body =  [NTYJSON parse:self.HTTPBody];
        NSLogInfo(@"self.HTTPBody:%@ url:%@",body,self.URL.absoluteString);
        NSLogInfo(@"self.HTTPBodyStream:%@",self.HTTPBodyStream);
        NSURLComponents *remoteURLComs = [NSURLComponents componentsWithString:[MYUserService shared].latestRemoteAccessURL];
        NSString *portStr = STRING(@"%@",[remoteURLComs port]);
        NSString *hostStr = remoteURLComs.host;
        [self setValue:tokenValue forHTTPHeaderField:@"z-token"];
        [self setValue:hostStr forHTTPHeaderField:@"Nas-Neigh-Ip"];
        [self setValue:portStr forHTTPHeaderField:@"Nas-Neigh-Port"];
        NSLogInfo(@"remoteAccessURL&appendheaders::%@\n%@",self.URL,self.allHTTPHeaderFields);
        return YES;
    }
    return NO;
}

@end
  • 接下来，在需要进行拦截的WebViewController中，注册MYSchemeURLProtocol。核心代码示例如下：
#import "MYWebviewController.h"
#import "MYSchemeURLProtocol.h"


@interface JKJWebviewController ()<WKNavigationDelegate,WKScriptMessageHandler,WKUIDelegate>

@property (nonatomic, strong) WKWebView      *webView;
@property (nonatomic, strong) NSURLRequest   *request;
@end

@implementation JKJWebviewController

- (void)dealloc {
    [[NSNotificationCenter defaultCenter] removeObserver:self];
#if JKJSchemeURLHeaderTokenAdd
    if(self.refModule.refModuleRequest == JKJRefModuleRequestRemoteAccessWebLoad){
        [JKJSchemeURLProtocol unregistSchemeURLProtocol];
    }
#endif
}


- (void)viewDidLoad {
    [super viewDidLoad];
//注册代码
#if JKJSchemeURLHeaderTokenAdd
    if(self.refModule.refModuleRequest == JKJRefModuleRequestRemoteAccessWebLoad){
        [JKJSchemeURLProtocol registSchemeURLProtocol];
    }
#endif
    
    // Do any additional setup after loading the view.
    self.title                  = self.webTitle;
    self.edgesForExtendedLayout = UIRectEdgeNone;
    
    WKWebViewConfiguration *configuration = [self configuration];
    WKWebView *webView = [[WKWebView alloc] initWithFrame:self.view.bounds configuration:configuration];
    webView.autoresizingMask           = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
    webView.navigationDelegate         = self;
    webView.UIDelegate = self;
    [webView.scrollView setShowsVerticalScrollIndicator:NO];
    [self.view addSubview:webView];


    if (self.needObserveTitle) {
        [webView addObserver:self forKeyPath:@"title" options:NSKeyValueObservingOptionNew context:NULL];
    }

    self.webView = webView;
    
    NSMutableURLRequest *req =  [[NSMutableURLRequest alloc] initWithURL:[NSURL fileURLWithPath:urlString]];

    [webView loadRequest:req];
    self.request = req;
}
  • 执行结果：
  1.  所有请求都可以进行拦截，
  2. 所有get请求结果拦截修改和加载表现正常
  3. post请求中的body体丢失， 这个问题暂时无法在所运用的场景中使用到，由于后期很多链接都有post请求转发，并且携带有body体，NSURLProtocol拦截之后body体丢失问题无法得到完整的解决，最后放弃了拦截修改。
  4. 暂未找到需求逻辑最佳解决方案，待后续处理补充。
xuewu1011
关注
  • 3
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
WKWebView实现请求拦截，http/https/file等
colrying的博客
08-17 6158
本文是通过WKURLSchemeHandler的方式实现的请求拦截，WKWebView的WKURLSchemeHandler是不支持拦截http/https/file等原始协议的，如果通过[config setURLSchemeHandler:handler forURLScheme:@"http"];就会闪退，但是支持自定义协议，但是有时候项目就是需要拦截这些请求，该怎么处理呢？ 一、自定义一个HFURLSchemeHandler类，并实现WKURLSchemeHandler代理 1、HFURLSch
ios-基于WKWebView的hybride的demo 。拦截了css,js,png等所有客户端请求。原理还是NSURLProtocol.zip
07-11
直接gittub 地址吧。 https://github.com/LiuShuoyu/HybirdWKWebVIew。
iOS和JS交互教程之WKWebView-协议拦截详解
01-04
前言 由于Xcode8发布之后，编译器开始不支持iOS 7了，这样我们的app也改为最低支持iOS 8.0，既然需要与web交互，那自然也就选择使用了 iOS 8.0之后 才推出的新控件 WKWebView. 相比与 UIWebView, WKWebView 存在很多优势： 支持更多的HTML5的特性 高达60fps滚动刷新频率与内置手势 与Safari相容的JavaScript引擎 在性能、稳定性方面有很大提升占用内存更少 协议方法及功能都更细致 可获取加载进度等。 先解释下标题：“iOS与JS交互”。iOS指iOS原生代码（文章只有OC示例），JS指WEB前端（不单指J
详解iOS开发之NSURLProtocol的那些坑
08-28
本篇文章主要介绍了详解iOS开发之NSURLProtocol的那些坑，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
WKWebview拦截导致http请求body为空解决方案
漂客的博客
05-06 4063
wkwebview正常是不允许拦截的，但是官方文档中给出了私有api： registerSchemeForCustomProtocol unregisterSchemeForCustomProtocol 具体的拦截方式，推荐看下此链接： https://www.jianshu.com/p/14e5156edc07 拦截后出现的问题1： .此拦截方式是私有api，上线可能被拒， ...
iOS 通过NSURLProtocol拦截WKWebView网络请求
最新发布
chengqiang0414的博客
02-27 1181
拦截WKWebView网络请求，替换本地资源或者更改请求
SSWKURL:Intercept Requests of WKWebView. WKWebView网络请求拦截
05-04
SSWKURL Intercept and cache requests of WKWebView. 介绍 网易团队采用了更完整细致(原理相同)的解决方案,建议采用他们的: Features Intercept All Requests of WKWebView. Cache requests. Installation Drag the SSWKURL.h and SSWKURL.m files into your project. Usage Subclass SSWKURLProtocol like NSURLProtol. Implement your own -startLoading: and -stopLoading. Then call [yourWKURLConfiguration ssRegisterURLProtocol:[YourSSWKURLProtocol cl
iOS_OC_拦截WKWebview的网络请求
qq_23577319的博客
08-07 3797
应用场景
swift WKWebView 实现 URLProtocol 网络拦截 修改 Referer添加
qq_40023759的博客
09-20 3587
在我们的日常开发过程中有时需要我们对网络做拦截操作，诸如对request 的header 做修改 添加等。 对于有效的url(http 或者 https) 只需要对request 做添加头文件即可 //这里是对资源加入的防盗链 var request = URLRequest(url:url as URL); request.setValue(kReferer, forHTTPHead...
iOS拦截 HTTP 请求demo源码：修改请求的HTTPHeaderField,设置代理IP和端口
04-14
对于 WKWebView 中发出的网络请求也无能为力`，如果真的要拦截来自 WKWebView 中的请求，还是需要实现 WKWebView 对应的 WKNavigationDelegate，并在代理方法中获取请求。 应用场景： 1、 自定义请求头的...
iOS使用 NSURLProtocol 拦截 HTTP 请求的简单例子
04-19
1、文章：https://kunnan.blog.csdn.net/article/details/115690756 2、原理：利用NSURLProtocol 拦截 HTTP 请求 3、应用场景：隧道APP请求我们自己接口的都不走隧道、修改请求的HTTPHeaderField,设置代理IP和端口、防抓包（使Thor，Charles，Burp等代理抓包方式全部失效）
ios wkwebview离线化加载h5资源解决方案
01-04
思路: 使用NSURLProtocol拦截请求转发到本地。 1.确认离线化需求 部门负责的app有一部分使用的线上h5页，长期以来加载略慢… 于是考虑使用离线化加载。 确保[低速网络]或[无网络]可网页秒开。 2.使用...
如何取消对WKWebView的请求拦截
qq_15509071的博客
04-19 1776
使用unregisterSchemeForCustomProtocol取消拦截 [NSURLProtocol unregisterClass:[MyURLProtocol class]]; Class cls = NSClassFromString(@"WKBrowsingContextController"); SEL sel = NSSelectorF...
关于WKWebView添加Cookie的处理以及拦截和重定向
HaapyWei
12-10 2624
//初始化添加了这个属性后，web页里点击后的请求都会有添加这个cookie，否则url请求以后便不带cookie WKUserContentController *userContentController = WKUserContentController.new; NSString * cookieSource = [NSString s...
强悍的NSURLProtocol （webview的重定向）
allanGold
12-22 1862
http://www.cocoachina.com/ios/20161202/18261.html
iOS 关于重定向的那些事(NSURLProcotol-WKWebView)
weixin_30362083的博客
07-31 1241
重定向定义：重定向(Redirect)就是通过各种方法将各种网络请求重新定个方向转到其它位置（如：网页重定向、域名的重定向、路由选择的变化也是对数据报文经由路径的一种重定向）。   NSURLProtocol：NSURLProtocol能够让你去重新定义苹果的URL加载系统 (URL Loading System)的行为，URL Loading System里有许多类用于处理UR...
iOS WebView 忽略不受信任的https证书（SSL）
WangQingLei0307的博客
03-21 4377
大家在开发过程中经常会使用到WebView，但是在使用WebView经常回家再Https，可是有一些Https是不受信任的，这个时候我们就要跳过这个Https信任这个过程，那我们如何跳过这个过程呢？ 为了方便使用，我们创建一个类别。 类别名字NSURLRequest+IgnoreSSL.h #import <Foundation/Foundation.h> NS_ASSUM...
NSURLProtocol
weixin_33896069的博客
06-16 110
由于UIWebView无法实现离线缓存，因此想利用Archieve机制来实现文件形式的离线缓存机制。同时，由于NSURLRequest每一次对链接的请求，都将触发NSURLProtocol的回调，因此对NSURLProtocol合理应用可以很好的达到离线缓存的目的。 一、NSURLProtocol与NSURLProtocolClient简介： 首先，我先介绍一下NSURLProtoco...
WKWebView实现NSURLProtocol
my blog
12-20 1808
代码转载自：https://github.com/LiuShuoyu/HybirdWKWebVIew
uniapp如何实现请求拦截和响应拦截步骤
06-02
Uniapp 实现请求拦截和响应拦截的步骤如下： 1. 在 `main.js` 中注册拦截器： ```js // 注册请求拦截器 uni.interceptors.request.use(function (config) { // 在发送请求之前做些什么 return config; }, function (error) { // 对请求错误做些什么 return Promise.reject(error); }); // 注册响应拦截器 uni.interceptors.response.use(function (response) { // 对响应数据做点什么 return response; }, function (error) { // 对响应错误做点什么 return Promise.reject(error); }); ``` 2. 在拦截器中进行业务操作，例如添加请求头、处理响应数据等。 3. 在请求时使用 `uni.request` 方法发送请求，拦截器会自动拦截请求并进行业务操作。 例如，在请求时需要添加 token，可以在拦截器中添加请求头： ```js // 注册请求拦截器 uni.interceptors.request.use(function (config) { // 在发送请求之前做些什么 const token = uni.getStorageSync('token'); if (token) { config.header['Authorization'] = `Bearer ${token}`; } return config; }, function (error) { // 对请求错误做些什么 return Promise.reject(error); }); ``` 这样，在发送请求时，就会自动带上 token。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值