Ticket #2963: nonblock-connect-test.cpp

/*
 * Copyright 2008, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Distributed under the terms of the MIT License.
 */

#include <errno.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/time.h>


#ifdef __HAIKU__
#   include <OS.h>
#else

typedef int64_t bigtime_t;

static bigtime_t
system_time()
{
    timeval tv;
    gettimeofday(&tv, NULL);
    return (bigtime_t)tv.tv_sec * 1000000 + tv.tv_usec;
}


#endif  // !__HAIKU__


int
main(int argc, const char* const* argv)
{
    // create socket
    int fd = socket(AF_INET, SOCK_STREAM, 0);
    if (fd < 0) {
        fprintf(stderr, "failed to create socket: %s\n", strerror(errno));
        exit(1);
    }

    // set to non-blocking
    int flags = fcntl(fd, F_GETFL);
    if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) != 0) {
        fprintf(stderr, "failed to set to non-blocking: %s\n",
            strerror(errno));
        exit(1);
    }

    bigtime_t startTime = system_time();

    // connect it
    sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(0x4a7d2768);
    addr.sin_port = htons(80);
    socklen_t addrLen = sizeof(addr);
    if (connect(fd, (sockaddr*)&addr, addrLen) < 0) {
        if (errno != EINPROGRESS) {
            fprintf(stderr, "failed to connect(): %s\n", strerror(errno));
            exit(1);
        }
    }

    // wait for the connection to be established
    fd_set writeSet;
    FD_ZERO(&writeSet);
    FD_SET(fd, &writeSet);

    int result = select(fd + 1, NULL, &writeSet, NULL, NULL);
    bigtime_t endTime = system_time();
    if (result != 1)
        fprintf(stderr, "select() unexpectedly returned: %d\n", result);

    printf("Connection took %llu us\n", endTime - startTime);

    return 0;
}