一、需求背景
NSURLProtocol可以攔截UIWebView/WKWebView里的請求，公司的產(chǎn)品需要將攔劫到的HTTPS請求的URL的Host改成我們自己特定的Host，然后在請求頭里帶上原始域名給我們的節(jié)點(diǎn)殉簸，節(jié)點(diǎn)再以原始域名去訪問源站。我們稱之為HTTPS域名收斂。
域名被收斂之后面臨著一個問題略就，就是網(wǎng)頁里的html/css/js等文件被收斂之后，里面很多用相對地址編碼的資源被WebView加載時域名也變成了我們那個特定域名晃酒，這樣這些資源的原始域名就丟失了表牢，導(dǎo)致WebView里很多資源加載失敗。問題的解決方法是在NSURLProtocol里做域名收斂收到NSHTTPURLResponse的時候贝次，將它的URL的Host還原成原來的域名崔兴。
二、修改NSHTTPURLResponse的URL
NSHTTPURLResponse的URL屬性是只讀的蛔翅，也沒有被KVC敲茄，我們無法簡單的去直接修改它。所以只能根據(jù)原來的Response重新再構(gòu)造出一個新的URL不一樣的Response山析。NSHTTPURLResponse初始化方法如下
- (nullable instancetype)initWithURL:(NSURL *)url
statusCode:(NSInteger)statusCode
HTTPVersion:(nullable NSString *)HTTPVersion
headerFields:(nullable NSDictionary<NSString *, NSString *> *)headerFields;
構(gòu)造NSHTTPURLResponse所需的其他參數(shù)都好說堰燎，唯獨(dú)HTTPVersion這個參數(shù)我們無法直接得到，也就是沒有從原來那個NSHTTPURLResponse獲取HTTPVersion的Public API或者Private API笋轨。
三秆剪、獲取NSURLResponse的HTTPVersion
查資料后知道NSURLResponse是基于_CFURLResponse這個結(jié)構(gòu)體實(shí)現(xiàn)的
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中獲取到這個結(jié)構(gòu)體
SEL selector = NSSelectorFromString(@"_CFURLResponse");
CFTypeRef response = CFBridgingRetain([response performSelector:selector]);
CFShow(response);
拿到CFURLResponseRef赊淑，又要怎么從它獲取到httpVersion呢？無意間又發(fā)現(xiàn)下面這個函數(shù)可以將CFURLResponseRef轉(zhuǎn)化為CFHTTPMessageRef
CFHTTPMessageRef CFURLResponseGetHTTPResponse(CFURLResponseRef);
而下面這個函數(shù)又可以從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的函數(shù)實(shí)現(xiàn)
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轉(zhuǎn)化為CFHTTPMessageRef
CFHTTPMessageRef message = originURLResponseGetHTTPResponse(cfResponse);
// 獲取http協(xié)議版本
CFStringRef cfVersion = CFHTTPMessageCopyVersion(message);
if (NULL != cfVersion) {
version = (__bridge NSString *)cfVersion;
CFRelease(cfVersion);
}
CFRelease(cfResponse);
}
}
// 獲取失敗的話則設(shè)置一個默認(rèn)值
if (nil == version || 0 == version.length) {
version = @"HTTP/1.1";
}
return version;
}
@end
五膏燃、最后的話
- _CFURLResponse和CFURLResponseGetHTTPResponse都是蘋果沒有公開的，使用時需要特殊處理下何什，以防上架時被蘋果判定為使用了Private API而被拒组哩。至于怎么處理就不細(xì)說了，可以采用拼接字符串或者對其進(jìn)行加解密的方式处渣。
- 之前做HTTP2的時候伶贰，不知道在客戶端要怎樣獲取請求使用的HTTP協(xié)議版本，以判斷是否協(xié)商使用了HTTP2協(xié)議罐栈，都是通過在服務(wù)端查看日志來判斷的∈蜓茫現(xiàn)在好了，直接通過NSURLResponse就可以獲取到了荠诬。
- 說來慚愧琅翻，工作多年了還是頭一回寫博客。哪里寫得不好或者有什么錯誤柑贞，還請大家多多包涵并給予指正方椎。
