tencent cloud

// Register `NSURLProtocol` to intercept requests
[NSURLProtocol registerClass:[MSDKDnsHttpMessageTools class]];

// Set the SNI URL
NSString *originalUrl = @"your url";
NSURL *url = [NSURL URLWithString:originalUrl];
NSMutableURLRequest *request = [[NSMutableURLRequest alloc] initWithURL:url];
NSArray *result = [[MSDKDns sharedInstance] WGGetHostByName:url.host];
NSString *ip = nil;
if (result && result.count > 1) {
    if (![result[1] isEqualToString:@"0"]) {
        ip = result[1];
    } else {
        ip = result[0];
    }
}
// Get the IP successfully through HTTPDNS, replace the URL, and set the host header
if (ip) {
    NSRange hostFirstRange = [originalUrl rangeOfString:url.host];
    if (NSNotFound != hostFirstRange.location) {
        NSString *newUrl = [originalUrl stringByReplacingCharactersInRange:hostFirstRange withString:ip];
        request.URL = [NSURL URLWithString:newUrl];
        [request setValue:url.host forHTTPHeaderField:@"host"];
    }
}

// Sample `NSURLConnection`
self.connection = [[NSURLConnection alloc] initWithRequest:request delegate:self];
[self.connection start];

// Sample `NSURLSession`
NSURLSessionConfiguration *configuration = [NSURLSessionConfiguration defaultSessionConfiguration];
NSArray *protocolArray = @[ [MSDKDnsHttpMessageTools class] ];
configuration.protocolClasses = protocolArray;
NSURLSession *session = [NSURLSession sessionWithConfiguration:configuration delegate:self delegateQueue:[NSOperationQueue mainQueue]];
self.task = [session dataTaskWithRequest:request];
[self.task resume];

// Note: When you use `NSURLProtocol` to intercept a POST request made by `NSURLSession`, `HTTPBody` will be empty.
// There are two solutions:
// 1. Use `NSURLConnection` to send POST requests.
// 2. Place `HTTPBody` in the HTTP header field first and then get it from `NSURLProtocol`.
// The following mainly demonstrates the second solution
// NSString *postStr = [NSString stringWithFormat:@"param1=%@&param2=%@", @"val1", @"val2"];
// [_request addValue:postStr forHTTPHeaderField:@"originalBody"];
// _request.HTTPMethod = @"POST";
// NSURLSessionConfiguration *configuration = [NSURLSessionConfiguration defaultSessionConfiguration];
// NSArray *protocolArray = @[ [CFHttpMessageURLProtocol class] ];
// configuration.protocolClasses = protocolArray;
// NSURLSession *session = [NSURLSession sessionWithConfiguration:configuration delegate:self delegateQueue:[NSOperationQueue mainQueue]];
// NSURLSessionTask *task = [session dataTaskWithRequest:_request];
// [task resume];

#pragma mark - In SNI scenarios, call it only once
/**
 Set the list of domains to be intercepted in SNI scenarios
 We recommend you call the API to intercept domains only in SNI scenarios but not other scenarios

 @param hijackDomainArray List of domains to be intercepted
 */
- (void) WGSetHijackDomainArray:(NSArray *)hijackDomainArray;

/**
 Set the list of domains not to be intercepted in SNI scenarios

 @param noHijackDomainArray List of domains not to be intercepted
 */
 - (void) WGSetNoHijackDomainArray:(NSArray *)noHijackDomainArray;