一、需求背景
NSURLProtocol可以拦截UIWebView/WKWebView里的请求，公司的产品需要将拦劫到的HTTPS请求的URL的Host改成我们自己特定的Host，然后在请求头里带上原始域名给我们的节点，节点再以原始域名去访问源站。我们称之为HTTPS域名收敛。
域名被收敛之后面临着一个问题，就是网页里的html/css/js等文件被收敛之后，里面很多用相对地址编码的资源被WebView加载时域名也变成了我们那个特定域名，这样这些资源的原始域名就丢失了，导致WebView里很多资源加载失败。问题的解决方法是在NSURLProtocol里做域名收敛收到NSHTTPURLResponse的时候，将它的URL的Host还原成原来的域名。
二、修改NSHTTPURLResponse的URL
NSHTTPURLResponse的URL属性是只读的，也没有被KVC，我们无法简单的去直接修改它。所以只能根据原来的Response重新再构造出一个新的URL不一样的Response。NSHTTPURLResponse初始化方法如下
- (nullable instancetype)initWithURL:(NSURL *)url
statusCode:(NSInteger)statusCode
HTTPVersion:(nullable NSString *)HTTPVersion
headerFields:(nullable NSDictionary<NSString *, NSString *> *)headerFields;
构造NSHTTPURLResponse所需的其他参数都好说，唯独HTTPVersion这个参数我们无法直接得到，也就是没有从原来那个NSHTTPURLResponse获取HTTPVersion的Public API或者Private API。
三、获取NSURLResponse的HTTPVersion
查资料后知道NSURLResponse是基于_CFURLResponse这个结构体实现的
typedef struct _CFURLResponse {
CFRuntimeBase _base;
CFAbsoluteTime creationTime;
CFURLRef url;
CFStringRef mimeType;
int64_t expectedLength;
CFStringRef textEncoding;
CFIndex statusCode;
CFStringRef httpVersion;
CFDictionaryRef headerFields;
Boolean isHTTPResponse;
OSSpinLock parsedHeadersLock;
ParsedHeaders* parsedHeaders;
} _CFURLResponse;
typedef const struct _CFURLResponse* CFURLResponseRef;
你可以通过以下代码从NSURLResponse中获取到这个结构体
SEL selector = NSSelectorFromString(@"_CFURLResponse");
CFTypeRef response = CFBridgingRetain([response performSelector:selector]);
CFShow(response);
拿到CFURLResponseRef，又要怎么从它获取到httpVersion呢？无意间又发现下面这个函数可以将CFURLResponseRef转化为CFHTTPMessageRef
CFHTTPMessageRef CFURLResponseGetHTTPResponse(CFURLResponseRef);
而下面这个函数又可以从CFHTTPMessageRef中获取到我们想要的HttpVersion
CFStringRef CFHTTPMessageCopyVersion(CFHTTPMessageRef message);
四、示例代码
#import <dlfcn.h>
#import "NSURLResponse+Help.h"
@implementation NSURLResponse (Help)
typedef CFHTTPMessageRef (*MYURLResponseGetHTTPResponse)(CFURLRef response);
- (NSString *)getHTTPVersion {
NSURLResponse *response = self;
NSString *version;
// 获取CFURLResponseGetHTTPResponse的函数实现
NSString *funName = @"CFURLResponseGetHTTPResponse";
MYURLResponseGetHTTPResponse originURLResponseGetHTTPResponse =
dlsym(RTLD_DEFAULT, [funName UTF8String]);
SEL theSelector = NSSelectorFromString(@"_CFURLResponse");
if ([response respondsToSelector:theSelector] &&
NULL != originURLResponseGetHTTPResponse) {
// 获取NSURLResponse的_CFURLResponse
CFTypeRef cfResponse = CFBridgingRetain([response performSelector:theSelector]);
if (NULL != cfResponse) {
// 将CFURLResponseRef转化为CFHTTPMessageRef
CFHTTPMessageRef message = originURLResponseGetHTTPResponse(cfResponse);
// 获取http协议版本
CFStringRef cfVersion = CFHTTPMessageCopyVersion(message);
if (NULL != cfVersion) {
version = (__bridge NSString *)cfVersion;
CFRelease(cfVersion);
}
CFRelease(cfResponse);
}
}
// 获取失败的话则设置一个默认值
if (nil == version || 0 == version.length) {
version = @"HTTP/1.1";
}
return version;
}
@end
五、最后的话
- _CFURLResponse和CFURLResponseGetHTTPResponse都是苹果没有公开的，使用时需要特殊处理下，以防上架时被苹果判定为使用了Private API而被拒。至于怎么处理就不细说了，可以采用拼接字符串或者对其进行加解密的方式。
- 之前做HTTP2的时候，不知道在客户端要怎样获取请求使用的HTTP协议版本，以判断是否协商使用了HTTP2协议，都是通过在服务端查看日志来判断的。现在好了，直接通过NSURLResponse就可以获取到了。
- 说来惭愧，工作多年了还是头一回写博客。哪里写得不好或者有什么错误，还请大家多多包涵并给予指正。
