iOS WebView 中的 Cookie场景IP直连的几种方法（4）

iOS11 之前的处理方法：NSURLProtocal拦截后，手动管理 Cookie 的存储：


步骤：做 IP 替换时将原始 URL 保存到 Header 中

2. + (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {
3.     NSMutableURLRequest *mutableReq = [request mutableCopy];
4.     NSString *originalUrl = mutableReq.URL.absoluteString;
5.     NSURL *url = [NSURL URLWithString:originalUrl];
6.     // 异步接口获取IP地址
7.     NSString *ip = [[HttpDnsService sharedInstance] getIpByHostAsync:url.host];
8.     if (ip) {
9.         NSRange hostFirstRange = [originalUrl rangeOfString:url.host];
10.         if (NSNotFound != hostFirstRange.location) {
11.             NSString *newUrl = [originalUrl stringByReplacingCharactersInRange:hostFirstRange withString:ip];
12.             mutableReq.URL = [NSURL URLWithString:newUrl];
13.             [mutableReq setValue:url.host forHTTPHeaderField:@"host"];
14.             // 添加originalUrl保存原始URL
15.             [mutableReq addValue:originalUrl forHTTPHeaderField:@"originalUrl"];
16.         }
17.     }
18.     NSURLRequest *postRequestIncludeBody = [mutableReq cyl_getPostRequestIncludeBody];
19.     return postRequestIncludeBody;
20. }

1. - (void)handleCookiesFromResponse:(NSURLResponse *)response {
2.     NSString *originalURLString = [self.request valueForHTTPHeaderField:@"originalUrl"];
3.     if ([response isKindOfClass:[NSHTTPURLResponse class]]) {
4.         NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)response;
5.         NSDictionary<NSString *, NSString *> *allHeaderFields = httpResponse.allHeaderFields;
6.         if (originalURLString && originalURLString.length > 0) {
7.             NSArray *cookies = [NSHTTPCookie cookiesWithResponseHeaderFields:allHeaderFields forURL: [[NSURL alloc] initWithString:originalURLString]];
8.             if (cookies && cookies.count > 0) {
9.                 NSURL *originalURL = [NSURL URLWithString:originalURLString];
10.                 [[NSHTTPCookieStorage sharedHTTPCookieStorage] setCookies:cookies forURL:originalURL mainDocumentURL:nil];
11.             }
12.         }
13.     }
14. }
16. - (void)task:(CFHTTPDNSRequestTask *)task didReceiveResponse:(NSURLResponse *)response cachePolicy:(NSURLCacheStoragePolicy)cachePolicy {
17.     [self handleCookiesFromResponse:response];
18.     [self.client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:cachePolicy];
19. }

1. + (void)handleCookieWithRequest:(NSMutableURLRequest *)request {
2.     NSString* originalURLString = [request valueForHTTPHeaderField:@"originalUrl"];
3.     if (!originalURLString || originalURLString.length == 0) {
4.         return;
5.     }
6.     NSArray *cookies = [[NSHTTPCookieStorage sharedHTTPCookieStorage] cookies];
7.     if (cookies && cookies.count >0) {
8.         NSDictionary *cookieHeaders = [NSHTTPCookie requestHeaderFieldsWithCookies:cookies];
9.         NSString *cookieString = [cookieHeaders objectForKey:@"Cookie"];
10.         [request addValue:cookieString forHTTPHeaderField:@"Cookie"];
11.     }
12. }
14. + (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {
15.     NSMutableURLRequest *mutableReq = [request mutableCopy];
16. //...
17.      [self handleCookieWithRequest:mutableReq];
18.     return [mutableReq copy];
19. }