获取NSURLResponse的HTTPVersion

一、需求背景

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 *)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 
#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

五、最后的话

1. _CFURLResponse和CFURLResponseGetHTTPResponse都是苹果没有公开的，使用时需要特殊处理下，以防上架时被苹果判定为使用了Private API而被拒。至于怎么处理就不细说了，可以采用拼接字符串或者对其进行加解密的方式。
2. 之前做HTTP2的时候，不知道在客户端要怎样获取请求使用的HTTP协议版本，以判断是否协商使用了HTTP2协议，都是通过在服务端查看日志来判断的。现在好了，直接通过NSURLResponse就可以获取到了。
3. 说来惭愧，工作多年了还是头一回写博客。哪里写得不好或者有什么错误，还请大家多多包涵并给予指正。