让 WKWebView 支持 NSURLProtocol

WKWebView在github上源码地址:

https://github.com/WebKit/webkit/tree/master/Source/WebKit/UIProcess/API/Cocoa

TestProtocol.mm 中看到了 NSURLProtocol 熟悉的身影:

+ ( void)registerWithScheme:( NSString *)scheme
{
testScheme = [scheme  retain];
[ NSURLProtocol registerClass:[ self  class]];
#if WK_API_ENABLED
[ WKBrowsingContextController registerSchemeForCustomProtocol:testScheme];
#endif

从 registerSchemeForCustomProtocol: 这个方法名来猜测,它的作用的应该是注册一个自定义的 scheme,这样对于 WebKit 进程的所有网络请求,都会先检查是否有匹配的 scheme,有的话再走主进程的 NSURLProtocol 这一套流程,猜测这么做可能是为了保证效率 (NSURLRequest 的 HTTPBody 属性在 WKWebView 中被忽略了应该也出于这个原因),毕竟 IPC 代价挺高的。后续翻 WebKit::CustomProtocolManager 和 WebKit::WebProcessPool 等相关源码也印证了这个猜想。

Class cls =  NSClassFromString( @"WKBrowsingContextController");
SEL sel =  NSSelectorFromString( @"registerSchemeForCustomProtocol:");
if ([( id)cls respondsToSelector:sel]) {
// 把 http 和 https 请求交给 NSURLProtocol 处理
[( id)cls performSelector:sel withObject: @"http"];
[( id)cls performSelector:sel withObject: @"https"];
}
// 这下 AwesomeURLProtocol 就可以用啦
[ NSURLProtocol registerClass:[AwesomeURLProtocol  class]];

关于私有 API现在 WKWebView 中的所有请求都可以被 NSURLProtocol 修改了:

按照 @sunnyxx 的总结,Apple 检查私有 API 的使用,大概会采取下面几种手段:

  • 是否 link 了私有 framework 或者公开 framework 中的私有符号,这可以防止开发者把私有 header 都 dump 出来供程序直接调用。
  • 同上,使用@selector(_private_sel)加上-performSelector:的方式直接调用私有 API。
  • 扫描所有符号,查看是否有继承自私有类,重载私有方法,方法名是否有重合。
  • 扫描所有string,看字符串常量段是否出现和私有 API 对应的。

而本文所介绍的方法,一共有两个地方使用了私有 API:

Class cls =  NSClassFromString( @"WKBrowsingContextController");
SEL sel =  NSSelectorFromString( @"registerSchemeForCustomProtocol:");

这两个地方都是通过反射的方式拿到了私有的 class/selector,对应上面的第四条。其中第二行那个还好说,因为 registerSchemeForCustomProtocol 这个名词看上去相当普通,如果把这种字符串也禁掉了的话会误伤一大票开发者,所以有风险的主要是 WKBrowsingContextController 这个字符串,要前缀有前缀,要 camel case 有 camel case,再跟私有 class 名撞车的话就跟可能被拒了。

那么怎样绕过这个字符串呢?查询 WKWebView.h 可以看到,有个方法 - browsingContextController 的方法名跟 WKBrowsingContextController 长得很像,通过 KVC 取出来(没错,KVC 不但可以取 property 取 ivar,还可以取无入参 selector 的返回值)发现它就是 WKBrowsingContextController 的一个实例,这样一来这个私有类就可以通过 KVC 的方式来得到了:

Class cls = [[[WKWebView new] valueForKey:@"browsingContextController"class];

比起粗暴地 NSClassFromString,使用 valueForKey 的方法安全了许多。当然,如果还有什么要担心的话,这些字符串也可以不明着写出来,只要运行时算出来就行,比如用 base64 编码啊,图片资源里藏一段啊,甚至通过服务器下发……既然到了这个程度,苹果的静态扫描就很难再 hold 住了。

猜你喜欢

转载自blog.csdn.net/mlcldh/article/details/79897513
今日推荐