ifish/Ifish/airkiss/JMAirKissConnection.m

//
//  JMAirKissConnection.m
//  JMAirKiss
//
//  Created by shengxiao on 16/3/2.
//  Copyright © 2016年 shengxiao. All rights reserved.
//

#import "JMAirKissConnection.h"
#import "JMAirKissEncoder.h"
#import "GCDAsyncUdpSocket.h"
#import "ESPTouchTaskParameter.h"
#import "ESP_NetUtil.h"

#define kAirKiss_Port                    10000
#define kAirKiss_Host                    @"255.255.255.255"
#define kAirKiss_Limit_Return_Random_Num 20

@interface JMAirKissConnection()<GCDAsyncUdpSocketDelegate>
{
    JMAirKissEncoder  *_airKissEncoder;
    NSTimer           *_timer;          // 超过1分钟未连接成功则表示失败
    
    GCDAsyncUdpSocket *_clientUdpSocket;
    GCDAsyncUdpSocket *_serverUdpSocket;
    
    long              _tag;
    int               _returnRandomNum;
    
    BOOL              _connectionDone;
}
@property (nonatomic,strong) ESPTaskParameter *_parameter;
@end

@implementation JMAirKissConnection

- (instancetype)init
{
    self = [super init];
    if (self) {
        _airKissEncoder  = [[JMAirKissEncoder alloc] init];
        _tag             = 0;
        _returnRandomNum = 0;
        _connectionDone  = false;
        self._parameter = [[ESPTaskParameter alloc]init];
        
        // check whether IPv4 and IPv6 is supported
        NSString *localInetAddr4 = [ESP_NetUtil getLocalIPv4];
        if (![ESP_NetUtil isIPv4PrivateAddr:localInetAddr4]) {
            localInetAddr4 = nil;
        }
        NSString *localInetAddr6 = [ESP_NetUtil getLocalIPv6];
        [self._parameter setIsIPv4Supported:localInetAddr4!=nil];
        [self._parameter setIsIPv6Supported:localInetAddr6!=nil];
        
        [self setupClientUdpSocket];
        [self setupServerUdpSocket];
    }
    return self;
}

#pragma mark - Connection
/**
 *  AirKiss连接
 *
 *  @param ssidStr ssid
 *  @param pswStr  psw
 */
- (void)connectAirKissWithSSID:(NSString *)ssidStr
                      password:(NSString *)password {
    NSMutableArray *dataArray = [_airKissEncoder createAirKissEncorderWithSSID:ssidStr ? :@""
                                                                      password:password ? :@""];

    _tag                      = 0;
    _returnRandomNum          = 0;
    _connectionDone           = false;
    
    dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
        dispatch_async(dispatch_get_main_queue(), ^{
            _timer = [NSTimer scheduledTimerWithTimeInterval:60
                                                      target:self
                                                    selector:@selector(connectFailure)
                                                    userInfo:nil
                                                     repeats:NO];
        });
        
        for (int i = 0;i < dataArray.count;i++) {
            if (_connectionDone == true) {
                break;
            }
            UInt16   length      = [dataArray[i] unsignedShortValue];
            NSMutableData *mData = [NSMutableData data];
            UInt8    value       = 0;
            for (int j = 0; j < length; j++) {
                [mData appendBytes:&value length:1];
            }
            [_clientUdpSocket sendData:mData
                                toHost:kAirKiss_Host
                                  port:kAirKiss_Port
                           withTimeout:-1
                                   tag:_tag];
            
            [NSThread sleepForTimeInterval:0.004];
            
            _tag++;
        }
    });
}

- (void)closeConnection {
    _connectionDone = true;
    
    [_timer invalidate];
    _timer = nil;
    
    [_clientUdpSocket close];
    [_serverUdpSocket close];
    
    _clientUdpSocket = nil;
    _serverUdpSocket = nil;
}

#pragma mark - Set up udp socket
- (void)setupClientUdpSocket
{
    NSError *error = nil;
    
    if (!_clientUdpSocket) {
        _clientUdpSocket = [[GCDAsyncUdpSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];
        [_clientUdpSocket enableBroadcast:YES error:&error];
    }
    
    if (![_clientUdpSocket bindToPort:0 error:&error])
    {
        return;
    }
    
    if (![_clientUdpSocket beginReceiving:&error])
    {
        return;
    }
}

- (void)setupServerUdpSocket {
    if (!_serverUdpSocket) {
        _serverUdpSocket = [[GCDAsyncUdpSocket alloc] initWithDelegate:self delegateQueue:dispatch_get_main_queue()];
        [_serverUdpSocket enableBroadcast:YES error:nil];
    }
    
    NSError *error = nil;
    
    if (![_serverUdpSocket bindToPort:kAirKiss_Port error:&error])
    {
        return;
    }
    
    if (![_serverUdpSocket beginReceiving:&error])
    {
        return;
    }
}

#pragma mark - Event Response
- (void)connectFailure {
    [_timer invalidate];
    _timer          = nil;
    
    _connectionDone = true;
    
    if (_connectionFailure) {
        _connectionFailure();
    }
}

#pragma mark - GCDAsyncUdpSocketDelegate
- (void)udpSocket:(GCDAsyncUdpSocket *)sock didSendDataWithTag:(long)tag
{
    // You could add checks here
}

- (void)udpSocket:(GCDAsyncUdpSocket *)sock didNotSendDataWithTag:(long)tag dueToError:(NSError *)error
{
    // You could add checks here
}

- (void)udpSocket:(GCDAsyncUdpSocket *)sock
   didReceiveData:(NSData *)data
      fromAddress:(NSData *)address
withFilterContext:(id)filterContext
{
    if (_serverUdpSocket == sock) {
        if (_connectionDone) {
            return;
        }
        
        // 设备连接WIFI成功后会像10000端口发送至少20个UDP广播包所附带的随机数
        if (data != nil) {
            UInt8 *bytes = (UInt8 *) [data bytes];
            if (bytes[0] == _airKissEncoder.randomChar) {
                _returnRandomNum ++;
                
                if (_returnRandomNum >= 2) {
                    // 成功
                    [_timer invalidate];
                    _timer = nil;
                    
                    if (_returnRandomNum == kAirKiss_Limit_Return_Random_Num) {
                        _connectionDone = true;
                        NSData*MacData = [data subdataWithRange:NSMakeRange(1, data.length-1)];
                        NSString*macstring = [self.class dataToHexString:MacData];
                        if (_connectionSuccess) {
                            _connectionSuccess(macstring);
                        }
                    }
                }
            }
        }
    }
}
// NSdata进制转十六进制字符串
+(NSString *) dataToHexString:(NSData*)data
{
    NSUInteger len = [data length];
    char * chars = (char *)[data bytes];
    NSMutableString * hexString = [[NSMutableString alloc] init];
    
    for(NSUInteger i = 0; i < len; i++ )
        [hexString appendString:[NSString stringWithFormat:@"%0.2hhx", chars[i]]];
    return hexString;
}

// 十六进制字符串转NSdata进制
+(NSData *) stringToHexData:(NSString*)string{
    
    NSInteger len = [string length] / 2;    // Target length
    unsigned char *buf = malloc(len);
    unsigned char *whole_byte = buf;
    char byte_chars[3] = {'\0','\0','\0'};
    
    int i;
    for (i=0; i < [string length] / 2; i++) {
        byte_chars[0] = [string characterAtIndex:i*2];
        byte_chars[1] = [string characterAtIndex:i*2+1];
        *whole_byte = strtol(byte_chars, NULL, 16);
        whole_byte++;
    }
    NSData *data = [NSData dataWithBytes:buf length:len];
    free( buf );
    return data;

}
@end