最近在項目里由于電信那邊發生dns發生域名劫持，因此需要手動將URL請求的域名重定向到指定的IP地址，但是由于請求可能是通過NSURLConnection,NSURLSession或者AFNetworking等方式，因此要想統一進行處理，一開始是想通過Method Swizzling去hook cfnetworking底層方法，后來發現其實有個更好的方法--NSURLProtocol。
NSURLProtocol
NSURLProtocol能夠讓你去重新定義蘋果的URL加載系統 (URL Loading System)的行為，URL Loading System里有許多類用于處理URL請求，比如NSURL，NSURLRequest，NSURLConnection和NSURLSession等，當URL Loading System使用NSURLRequest去獲取資源的時候，它會創建一個NSURLProtocol子類的實例，你不應該直接實例化一個NSURLProtocol，NSURLProtocol看起來像是一個協議，但其實這是一個類，而且必須使用該類的子類，并且需要被注冊。
使用場景
不管你是通過UIWebView, NSURLConnection 或者第三方庫 (AFNetworking, MKNetworkKit等)，他們都是基于NSURLConnection或者 NSURLSession實現的，因此你可以通過NSURLProtocol做自定義的操作。
- 重定向網絡請求
- 忽略網絡請求，使用本地緩存
- 自定義網絡請求的返回結果
- 一些全局的網絡請求設置
攔截網絡請求
子類化NSURLProtocol并注冊
@interface CustomURLProtocol : NSURLProtocol
@end
然后在application:didFinishLaunchingWithOptions:方法中注冊該CustomURLProtocol，一旦注冊完畢后，它就有機會來處理所有交付給URL Loading system的網絡請求。
- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {
//注冊protocol
[NSURLProtocol registerClass:[CustomURLProtocol class]];
return YES;
}
實現CustomURLProtocol
注冊好了之后，現在可以開始實現NSURLProtocol的一些方法：
- +canInitWithRequest:
這個方法主要是說明你是否打算處理對應的request，如果不打算處理，返回NO，URL Loading System會使用系統默認的行為去處理；如果打算處理，返回YES，然后你就需要處理該請求的所有東西，包括獲取請求數據并返回給 URL Loading System。網絡數據可以簡單的通過NSURLConnection去獲取，而且每個NSURLProtocol對象都有一個NSURLProtocolClient實例，可以通過該client將獲取到的數據返回給URL Loading System。
這里有個需要注意的地方，想象一下，當你去加載一個URL資源的時候，URL Loading System會詢問CustomURLProtocol是否能處理該請求，你返回YES，然后URL Loading System會創建一個CustomURLProtocol實例然后調用NSURLConnection去獲取數據，然而這也會調用URL Loading System，而你在+canInitWithRequest:中又總是返回YES，這樣URL Loading System又會創建一個CustomURLProtocol實例導致無限循環。我們應該保證每個request只被處理一次，可以通過+setProperty:forKey:inRequest:標示那些已經處理過的request，然后在+canInitWithRequest:中查詢該request是否已經處理過了，如果是則返回NO。
+ (BOOL)canInitWithRequest:(NSURLRequest *)request
{
//只處理http和https請求
NSString *scheme = [[request URL] scheme];
if ( ([scheme caseInsensitiveCompare:@"http"] == NSOrderedSame ||
[scheme caseInsensitiveCompare:@"https"] == NSOrderedSame))
{
//看看是否已經處理過了，防止無限循環
if ([NSURLProtocol propertyForKey:URLProtocolHandledKey inRequest:request]) {
return NO;
}
return YES;
}
return NO;
}
- +canonicalRequestForRequest:
通常該方法你可以簡單的直接返回request，但也可以在這里修改request，比如添加header，修改host等，并返回一個新的request，這是一個抽象方法，子類必須實現。
+ (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
NSMutableURLRequest *mutableReqeust = [request mutableCopy];
mutableReqeust = [self redirectHostInRequset:mutableReqeust];
return mutableReqeust;
}
+(NSMutableURLRequest*)redirectHostInRequset:(NSMutableURLRequest*)request
{
if ([request.URL host].length == 0) {
return request;
}
NSString *originUrlString = [request.URL absoluteString];
NSString *originHostString = [request.URL host];
NSRange hostRange = [originUrlString rangeOfString:originHostString];
if (hostRange.location == NSNotFound) {
return request;
}
//定向到bing搜索主頁
NSString *ip = @"cn.bing.com";
// 替換域名
NSString *urlString = [originUrlString stringByReplacingCharactersInRange:hostRange withString:ip];
NSURL *url = [NSURL URLWithString:urlString];
request.URL = url;
return request;
}
- +requestIsCacheEquivalent:toRequest:
主要判斷兩個request是否相同，如果相同的話可以使用緩存數據，通常只需要調用父類的實現。
+ (BOOL)requestIsCacheEquivalent:(NSURLRequest *)a toRequest:(NSURLRequest *)b
{
return [super requestIsCacheEquivalent:a toRequest:b];
}
- -startLoading -stopLoading
這兩個方法主要是開始和取消相應的request，而且需要標示那些已經處理過的request。
- (void)startLoading
{
NSMutableURLRequest *mutableReqeust = [[self request] mutableCopy];
//標示改request已經處理過了，防止無限循環
[NSURLProtocol setProperty:@YES forKey:URLProtocolHandledKey inRequest:mutableReqeust];
self.connection = [NSURLConnection connectionWithRequest:mutableReqeust delegate:self];
}
- (void)stopLoading
{
[self.connection cancel];
}
- NSURLConnectionDataDelegate方法
在處理網絡請求的時候會調用到該代理方法，我們需要將收到的消息通過client返回給URL Loading System。
- (void) connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
[self.client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
}
- (void) connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
[self.client URLProtocol:self didLoadData:data];
}
- (void) connectionDidFinishLoading:(NSURLConnection *)connection {
[self.client URLProtocolDidFinishLoading:self];
}
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
[self.client URLProtocol:self didFailWithError:error];
}
現在你已經可以截取request并做你想做的事了，這里有個demo可以參考一下，截取request并重新定向到新的地址，具體dns解析方法可以參看DNS解析) ，如有不對，歡迎指正，哈～（有遇到iOS8 hook sdwebimage會發起多次請求，可以看下底下的評論）
