是否可以使用GCD调度源在套接字上执行异步connect()？

Question

您可以使用GCD调度源从套接字读取和写入,监视侦听套接字的传入连接,但我无法弄清楚如何使用调度源来连接套接字？

在伪代码中,它看起来像这样:

dispatch_source_t source = dispatch_source_create(DISPATCH_SOURCE_TYPE_WRITE, connectingSocket, ...);
dispatch_source_set_event_handler(source, ^{
  // Socket did connect or not
});
fcntl(connectingSocket, F_SETFL, O_NONBLOCK);
connect(connectingSocket, addr, len);
dispatch_source_resume(source);

这比使用更好select().

Answer 1

我最初误解了你的问题...抱歉.我现在得到它...你想要EINPROGRESS来自connect并有一个调度源通知你当connect呼叫需要注意而不是轮询select...这很容易破解,似乎工作:

#import <sys/types.h>
#import <sys/socket.h>
#import <netinet/in.h>
#import <arpa/inet.h>

@implementation AppDelegate
{
    dispatch_source_t foo;
}

- (void)applicationDidFinishLaunching:(NSNotification *)aNotification {

    int socketFD = socket(PF_INET, SOCK_STREAM, 0);
    if (socketFD == -1)
    {
        socketFD = -1;
        abort();
    }

    int flags = fcntl(socketFD, F_GETFL, 0);
    int status = fcntl(socketFD, F_SETFL, flags | O_NONBLOCK);
    if (status == -1)
    {
        close(socketFD);
        socketFD = -1;
        abort();
    }

    struct sockaddr_in sockaddr4 = {0};
    sockaddr4.sin_len         = sizeof(sockaddr4);
    sockaddr4.sin_family      = AF_INET;
    sockaddr4.sin_port        = htons(22);
    inet_aton("127.0.0.1", &sockaddr4.sin_addr);

    foo = dispatch_source_create(DISPATCH_SOURCE_TYPE_WRITE, socketFD, 0, dispatch_get_main_queue());
    dispatch_source_set_event_handler(foo, ^{
        if (connect(socketFD, (const struct sockaddr *)&sockaddr4, (socklen_t)sizeof(sockaddr4)))
        {
            int err = errno;
            NSLog(@"errno: %s", strerror(err));
            if (err == ECONNREFUSED)
            {
                abort();
            }
            else if (err == EISCONN)
            {
                // connected -- queue up other work
                DoStuff();

                // Cancel the source so it doesnt keep notifying...
                dispatch_source_cancel(foo);
            }
        }
    });

    dispatch_source_set_cancel_handler(foo, ^{
        NSLog(@"Cancel");
    });

    dispatch_resume(foo);

    // Do initial connect
    if (connect(socketFD, (const struct sockaddr *)&sockaddr4, (socklen_t)sizeof(sockaddr4)))
    {
        if(errno != EINPROGRESS)
        {
            close(socketFD);
            socketFD = -1;
            abort();
        }
    }
}

@end