266 lines
12 KiB
Objective-C
266 lines
12 KiB
Objective-C
//
|
||
// ResolvingEffectiveHostTest.m
|
||
// TrustHttpDNSTests
|
||
//
|
||
// Created by xuyecan on 2024/5/28.
|
||
// Copyright © 2024 trustapp.com. All rights reserved.
|
||
//
|
||
|
||
#import <Foundation/Foundation.h>
|
||
#import <stdatomic.h>
|
||
#import <mach/mach.h>
|
||
#import "HttpdnsService.h"
|
||
#import "HttpdnsRemoteResolver.h"
|
||
#import "TestBase.h"
|
||
|
||
@interface ResolvingEffectiveHostTest : TestBase<HttpdnsTTLDelegate>
|
||
|
||
@end
|
||
|
||
|
||
@implementation ResolvingEffectiveHostTest
|
||
|
||
+ (void)setUp {
|
||
[super setUp];
|
||
}
|
||
|
||
- (void)setUp {
|
||
[super setUp];
|
||
|
||
self.httpdns = [[HttpDnsService alloc] initWithAccountID:100000];
|
||
|
||
[self.httpdns setLogEnabled:YES];
|
||
[self.httpdns setReuseExpiredIPEnabled:NO];
|
||
|
||
[self.httpdns setTtlDelegate:self];
|
||
[self.httpdns setLogHandler:self];
|
||
|
||
self.currentTimeStamp = [[NSDate date] timeIntervalSince1970];
|
||
}
|
||
|
||
- (void)tearDown {
|
||
[super tearDown];
|
||
}
|
||
|
||
- (int64_t)httpdnsHost:(NSString *)host ipType:(TrustHttpDNS_IPType)ipType ttl:(int64_t)ttl {
|
||
// 为了在并发测试中域名快速过期，将ttl设置为随<E4B8BA><E99A8F>?-4<><34>?
|
||
return arc4random_uniform(4) + 1;
|
||
}
|
||
|
||
- (void)testNormalMultipleHostsResolve {
|
||
dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
|
||
|
||
// 使用正常解析到的ttl
|
||
[self.httpdns setTtlDelegate:nil];
|
||
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
[hostNameIpPrefixMap enumerateKeysAndObjectsUsingBlock:^(NSString * _Nonnull host, NSString * _Nonnull ipPrefix, BOOL * _Nonnull stop) {
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
XCTAssertGreaterThan(result.ttl, 0);
|
||
// 同步接口，不复用过期ip的情况下，解析出的ip一定是未过期的
|
||
XCTAssertLessThan([[NSDate date] timeIntervalSince1970], result.lastUpdatedTimeInterval + result.ttl);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
}];
|
||
dispatch_semaphore_signal(semaphore);
|
||
});
|
||
|
||
dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
|
||
}
|
||
|
||
- (void)testNonblockingMethodShouldNotBlockDuringMultithreadLongRun {
|
||
NSTimeInterval startTime = [[NSDate date] timeIntervalSince1970];
|
||
NSTimeInterval testDuration = 10;
|
||
int threadCountForEachType = 5;
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
long long executeStartTimeInMs = [[NSDate date] timeIntervalSince1970] * 1000;
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
long long executeEndTimeInMs = [[NSDate date] timeIntervalSince1970] * 1000;
|
||
// 非阻塞接口任何情况下不应该阻塞超<E5A19E><E8B685>?0ms
|
||
if (executeEndTimeInMs - executeStartTimeInMs >= 30) {
|
||
printf("XCTAssertWillFailed, host: %s, executeTime: %lldms\n", [host UTF8String], executeEndTimeInMs - executeStartTimeInMs);
|
||
}
|
||
XCTAssertLessThan(executeEndTimeInMs - executeStartTimeInMs, 30);
|
||
if (result) {
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
}
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
[NSThread sleepForTimeInterval:testDuration + 1];
|
||
}
|
||
|
||
- (void)testMultithreadAndMultiHostResolvingForALongRun {
|
||
NSTimeInterval startTime = [[NSDate date] timeIntervalSince1970];
|
||
NSTimeInterval testDuration = 10;
|
||
int threadCountForEachType = 4;
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
[self.httpdns resolveHostAsync:host byIpType:HttpdnsQueryIPTypeIpv4 completionHandler:^(HttpdnsResult *result) {
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
}];
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:host byIpType:HttpdnsQueryIPTypeIpv4];
|
||
if (result) {
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
}
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
sleep(testDuration + 1);
|
||
}
|
||
|
||
// 指定查询both，但域名都只配置了ipv4
|
||
// 这种情况下，会自动打标该域名无ipv6，后续的结果只会包含ipv4地址
|
||
- (void)testMultithreadAndMultiHostResolvingForALongRunBySpecifyBothIpv4AndIpv6 {
|
||
NSTimeInterval startTime = [[NSDate date] timeIntervalSince1970];
|
||
NSTimeInterval testDuration = 10;
|
||
int threadCountForEachType = 4;
|
||
|
||
// 计数时有并发冲突的可能，但只是测试，不用过于严谨
|
||
__block int syncCount = 0, asyncCount = 0, syncNonBlockingCount = 0;
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
HttpdnsResult *result = [self.httpdns resolveHostSync:host byIpType:HttpdnsQueryIPTypeBoth];
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue(!result.hasIpv6Address);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
|
||
syncCount++;
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
[self.httpdns resolveHostAsync:host byIpType:HttpdnsQueryIPTypeBoth completionHandler:^(HttpdnsResult *result) {
|
||
XCTAssertNotNil(result);
|
||
XCTAssertTrue(!result.hasIpv6Address);
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
|
||
asyncCount++;
|
||
}];
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
for (int i = 0; i < threadCountForEachType; i++) {
|
||
dispatch_async(dispatch_get_global_queue(0, 0), ^{
|
||
while ([[NSDate date] timeIntervalSince1970] - startTime < testDuration) {
|
||
NSString *host = [hostNameIpPrefixMap allKeys][arc4random_uniform((uint32_t)[hostNameIpPrefixMap count])];
|
||
NSString *ipPrefix = hostNameIpPrefixMap[host];
|
||
|
||
HttpdnsResult *result = [self.httpdns resolveHostSyncNonBlocking:host byIpType:HttpdnsQueryIPTypeBoth];
|
||
if (result) {
|
||
XCTAssertTrue([result.host isEqualToString:host]);
|
||
XCTAssertTrue(!result.hasIpv6Address);
|
||
NSString *firstIp = [result firstIpv4Address];
|
||
if (![firstIp hasPrefix:ipPrefix]) {
|
||
printf("XCTAssertWillFailed, host: %s, firstIp: %s, ipPrefix: %s\n", [host UTF8String], [firstIp UTF8String], [ipPrefix UTF8String]);
|
||
}
|
||
XCTAssertTrue([firstIp hasPrefix:ipPrefix]);
|
||
}
|
||
|
||
syncNonBlockingCount++;
|
||
[NSThread sleepForTimeInterval:0.1];
|
||
}
|
||
});
|
||
}
|
||
|
||
sleep(testDuration + 1);
|
||
|
||
int theoreticalCount = threadCountForEachType * (testDuration / 0.1);
|
||
|
||
// printf all the counts
|
||
printf("syncCount: %d, asyncCount: %d, syncNonBlockingCount: %d, theoreticalCount: %d\n", syncCount, asyncCount, syncNonBlockingCount, theoreticalCount);
|
||
}
|
||
|
||
@end
|