CocoaAsyncSocket (GCDAsyncSocket)適配IPv6

Auditore發表於2018-01-03

底層採用了第三方的socket庫:CocoaAsyncSocket,裡面包含了GCDAsyncSocket.h和GCDAsyncSocket.m檔案。閱讀原始碼可以發現,GCDAsyncSocket中已經對ipv4和ipv6同時做了支援,但是為何在ipv6情況下會connect失敗呢。根據程式碼執行過程可以發現,在方法 -  (BOOL)connectWithAddress4:(NSData *)address4 address6:(NSData *)address6 error:(NSError **)errPtr; 中 BOOL useIPv6變數被置為NO,從而導致程式碼不執行ipv6的建立操作,而是執行ipv4的建立,從而導致連線始終失敗。找到了問題的原因,下面就可以對其進行處理解決。對上文提到的方法進行修改。程式碼如下:

- (BOOL)connectWithAddress4:(NSData *)address4 address6:(NSData *)address6 error:(NSError **)errPtr
{
      LogTrace();
      NSAssert(dispatch_get_specific(IsOnSocketQueueOrTargetQueueKey), @"Must be dispatched on socketQueue");
      LogVerbose(@"IPv4: %@:%hu", [[self class] hostFromAddress:address4], [[self class] portFromAddress:address4]);
      LogVerbose(@"IPv6: %@:%hu", [[self class] hostFromAddress:address6], [[self class] portFromAddress:address6]);
 
    //增加的程式碼
    if(address6)
    {
        [self setIPv6Enabled:YES];
    }
……
}
複製程式碼

另外在setIPv6Enable方法中做一些修改,程式碼如下:

- (void)setIPv6Enabled:(BOOL)flag
{
      // Note:YES means kIPv6Disabled is OFF
     
      dispatch_block_t block = ^{
           
            if (flag)
                  config |= kPreferIPv6;   //修改後程式碼
            else
                  config |= kIPv6Disabled;
      };
     
      if (dispatch_get_specific(IsOnSocketQueueOrTargetQueueKey))
            block();
      else
            dispatch_async(socketQueue, block);
}
複製程式碼

然而,網上基本上所有人都是如上方法去適配,結果是絲毫不起。最後在一個以訛傳訛的文章下的評論裡,找到了正確答案。 GCDAsyncSocket.m中的 - (BOOL)connectWithAddress4:(NSData *)address4 address6:(NSData *)address6 error:(NSError **)errPtr;方法中加入

...
if (address6) {
[self setPreferIPv4OverIPv6:NO];
}
BOOL preferIPv6 = (config & kPreferIPv6) ? YES : NO;
...
複製程式碼

下面這段是關鍵 然後在·+ (NSMutableArray *)lookupHost:(NSString *)host port:(uint16_t)port error:(NSError **)errPtr;方法,找到else if (res->ai_family == AF_INET6)把方法體內全部替換為以下程式碼:

struct sockaddr_in6 *sockaddr = (struct sockaddr_in6 *)res->ai_addr;
in_port_t *portPtr = &sockaddr->sin6_port;
if ((portPtr != NULL) && (*portPtr == 0)) {
*portPtr = htons(port);
}
NSData *address6 = [NSData dataWithBytes:res->ai_addr length:res->ai_addrlen];
[addresses addObject:address6];
複製程式碼

相關文章