Update Tcping.m

paradiseduo · Sep 18, 2023 · 5963504 · 5963504
1 parent 1b84c48
commit 5963504
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tcping/Tcping.m b/tcping/Tcping.m
@@ -26,15 +26,15 @@ - (instancetype)initWith:(dispatch_group_t)group queue:(dispatch_queue_t)queue {
     return self;
 }
 
-- (void)connectSocket:(NSString *)domain port:(UInt16)port {
+- (void)connectSocket:(NSString *)domain port:(UInt16)port timeout:(NSTimeInterval)timeout {
     _domain = domain;
     _port = port;
     _socket = [[GCDAsyncSocket alloc] initWithDelegate:self delegateQueue:_queue];
     if (!_socket.isConnected) {
         dispatch_group_enter(_group);
         _startTime = [NSDate date];
         NSError * error = nil;
-        [_socket connectToHost:domain onPort:port withTimeout:1.0 error:&error];
+        [_socket connectToHost:domain onPort:port withTimeout:timeout error:&error];
         if (error) {
             dispatch_group_leave(_group);
         }
@@ -53,6 +53,7 @@ - (void)socketDidDisconnect:(GCDAsyncSocket *)sock withError:(NSError *)err {
     if (err) {
         [ConsoleIO writeMessage:[err localizedDescription] to:OutputTypeError];
         dispatch_group_leave(_group);
+        [sock disconnect];
     }
 }
 @end