由于最近項目要區(qū)分是否為移動端橱赠，因此需要手動將URL請求的域名重定向到指定的IP地址秩伞，但是由于請求可能是通過NSURLConnection,NSURLSession或者AFNetworking等方式，因此要想統(tǒng)一進行處理，一開始是想通過Method Swizzling去hook cfnetworking底層方法，后來發(fā)現(xiàn)其實有個更好的方法--NSURLProtocol宛渐。
NSURLProtocol
NSURLProtocol能夠讓你去重新定義蘋果的URL加載系統(tǒng) (URL Loading System)的行為，URL Loading System里有許多類用于處理URL請求司忱，比如NSURL皇忿，NSURLRequest，NSURLConnection和NSURLSession等坦仍，當URL Loading System使用NSURLRequest去獲取資源的時候，它會創(chuàng)建一個NSURLProtocol子類的實例叨襟，你不應該直接實例化一個NSURLProtocol繁扎，NSURLProtocol看起來像是一個協(xié)議，但其實這是一個類，而且必須使用該類的子類梳玫，并且需要被注冊爹梁。
使用場景
不管你是通過UIWebView, NSURLConnection 或者第三方庫 (AFNetworking, MKNetworkKit等)，他們都是基于NSURLConnection或者 NSURLSession實現(xiàn)的提澎，因此你可以通過NSURLProtocol做自定義的操作姚垃。
重定向網(wǎng)絡請求
忽略網(wǎng)絡請求，使用本地緩存
自定義網(wǎng)絡請求的返回結(jié)果
一些全局的網(wǎng)絡請求設置
攔截網(wǎng)絡請求
子類化NSURLProtocol并注冊
@interface CustomURLProtocol : NSURLProtocol
@end
然后在application:didFinishLaunchingWithOptions:方法中注冊該CustomURLProtocol盼忌，一旦注冊完畢后积糯，它就有機會來處理所有交付給URL Loading system的網(wǎng)絡請求。
- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {
//注冊protocol
[NSURLProtocol registerClass:[CustomURLProtocol class]];
return YES;
}
實現(xiàn)CustomURLProtocol
注冊好了之后谦纱，現(xiàn)在可以開始實現(xiàn)NSURLProtocol的一些方法：
+canInitWithRequest:
這個方法主要是說明你是否打算處理對應的request看成，如果不打算處理，返回NO跨嘉，URL Loading System會使用系統(tǒng)默認的行為去處理川慌；如果打算處理，返回YES祠乃，然后你就需要處理該請求的所有東西梦重，包括獲取請求數(shù)據(jù)并返回給 URL Loading System。網(wǎng)絡數(shù)據(jù)可以簡單的通過NSURLConnection去獲取亮瓷，而且每個NSURLProtocol對象都有一個NSURLProtocolClient實例琴拧，可以通過該client將獲取到的數(shù)據(jù)返回給URL Loading System。
這里有個需要注意的地方寺庄，想象一下艾蓝，當你去加載一個URL資源的時候，URL Loading System會詢問CustomURLProtocol是否能處理該請求斗塘，你返回YES赢织，然后URL Loading System會創(chuàng)建一個CustomURLProtocol實例然后調(diào)用NSURLConnection去獲取數(shù)據(jù)，然而這也會調(diào)用URL Loading System馍盟，而你在+canInitWithRequest:中又總是返回YES于置，這樣URL Loading System又會創(chuàng)建一個CustomURLProtocol實例導致無限循環(huán)。我們應該保證每個request只被處理一次贞岭，可以通過+setProperty:forKey:inRequest:標示那些已經(jīng)處理過的request八毯，然后在+canInitWithRequest:中查詢該request是否已經(jīng)處理過了，如果是則返回NO瞄桨。
+ (BOOL)canInitWithRequest:(NSURLRequest *)request
{
//只處理http和https請求
NSString *scheme = [[request URL] scheme];
if ( ([scheme caseInsensitiveCompare:@"http"] == NSOrderedSame ||
[scheme caseInsensitiveCompare:@"https"] == NSOrderedSame))
{
//看看是否已經(jīng)處理過了话速，防止無限循環(huán)
if ([NSURLProtocol propertyForKey:URLProtocolHandledKey inRequest:request]) {
return NO;
}
return YES;
}
return NO;
}
+canonicalRequestForRequest:
通常該方法你可以簡單的直接返回request，但也可以在這里修改request芯侥，比如添加header泊交，修改host等乳讥，并返回一個新的request，這是一個抽象方法廓俭，子類必須實現(xiàn)云石。
+ (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是否相同，如果相同的話可以使用緩存數(shù)據(jù)研乒，通常只需要調(diào)用父類的實現(xiàn)汹忠。
+ (BOOL)requestIsCacheEquivalent:(NSURLRequest *)a toRequest:(NSURLRequest *)b
{
return [super requestIsCacheEquivalent:a toRequest:b];
}
-startLoading -stopLoading
這兩個方法主要是開始和取消相應的request，而且需要標示那些已經(jīng)處理過的request雹熬。
- (void)startLoading
{
NSMutableURLRequest *mutableReqeust = [[self request] mutableCopy];
//標示改request已經(jīng)處理過了宽菜，防止無限循環(huán)
[NSURLProtocol setProperty:@YES forKey:URLProtocolHandledKey inRequest:mutableReqeust];
self.connection = [NSURLConnection connectionWithRequest:mutableReqeust delegate:self];
}
- (void)stopLoading
{
[self.connection cancel];
}
NSURLConnectionDataDelegate方法
在處理網(wǎng)絡請求的時候會調(diào)用到該代理方法，我們需要將收到的消息通過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];
}
現(xiàn)在你已經(jīng)可以截取request并做你想做的事了赋焕，這里有個demo可以參考一下，截取request并重新定向到新的地址仰楚，具體dns解析方法可以參看DNS解析) 隆判，如有不對，歡迎指正僧界，哈～
