#include "socket.h"
#include "log.h"

#include <string.h>
#include <stdio.h>
#include <assert.h>

#if defined(_WIN32)
static int InitializeWsa()
{
    WSADATA wsa;
    int rc;
    if ((rc = WSAStartup(MAKEWORD(2, 2), &wsa)) != 0)
    {
        LOG_ERROR("WSAStartup failed: %d", rc);
        return -1;
    }
    return 0;
}

#define close closesocket
#endif

int SocketConnect(const char *host, short port, int nonblocking)
{
    struct addrinfo hints, *servinfo = NULL, *p = NULL;
    int rv;
    char portstr[6];
    int sock;

#if defined(_WIN32)
    if (InitializeWsa() != 0)
    {
        return -1;
    }
#endif

    memset(&hints, 0, sizeof(hints));
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;

    assert(snprintf(portstr, sizeof(portstr), "%hu", port) < (int) sizeof(portstr));

    if ((rv = getaddrinfo(host, portstr, &hints, &servinfo)) != 0)
    {
        goto cleanup;
    }

    for (p = servinfo; p != NULL; p = p->ai_next)
    {
        if ((sock = socket(p->ai_family, p->ai_socktype,
                           p->ai_protocol)) == -1)
        {
            continue;
        }

        if (nonblocking)
        {
            SocketSetNonblocking(sock, 1);
        }

        if (connect(sock, p->ai_addr, p->ai_addrlen) == -1)
        {
            int err = SocketErrno;
            if (err == SOCKET_EINPROGRESS)
                break;
            close(sock);
            continue;
        }

        break;
    }

cleanup:

    if (servinfo)
    {
        freeaddrinfo(servinfo);
    }

    if (p == NULL)
    {
#if defined(_WIN32)
        WSACleanup();
#endif
        return -1;
    }

    return sock;
}

int SocketDisconnect(int sock)
{
    int rc = close(sock);
#if defined(_WIN32)
    WSACleanup();
#endif
    return rc;
}

int64_t SocketRecv(int sock, void *buf, size_t len, int flags)
{
    return recv(sock, buf, len, flags);
}

int64_t SocketSend(int sock, const void *buf, size_t len, int flags)
{
    return send(sock, buf, len, flags);
}

int SocketSendAll(int sock, const char *buf, size_t *len)
{
    size_t total = 0;
    int rv;
    size_t remaining = *len;

    while (remaining > 0)
    {
        if ((rv = send(sock, buf+total, remaining, 0)) == -1)
        {
            break;
        }
        total += rv;
        remaining -= rv;
    }

    *len = total;

    return rv == -1 ? -1 : 0;
}

int SocketSelect(int sock, int *events, int timeout)
{
    fd_set rfd, wfd;
    struct timeval tv;
    int rv;

    assert(sock != -1);
    assert(events != NULL);
    assert(*events != 0);

    FD_ZERO(&rfd);
    FD_ZERO(&wfd);

    if (*events & EV_READ)
    {
        FD_SET(sock, &rfd);
    }

    if (*events & EV_WRITE)
    {
        FD_SET(sock, &wfd);
    }

    memset(&tv, 0, sizeof(tv));
    tv.tv_sec = timeout / 1000;
    tv.tv_usec = (timeout - (tv.tv_sec * 1000)) * 1000;

    *events = 0;

    rv = select(sock+1, &rfd, &wfd, NULL, &tv);

    if (rv < 0)
    {
        return rv;
    }

    if (FD_ISSET(sock, &wfd))
    {
        *events = EV_WRITE;
    }

    if (FD_ISSET(sock, &rfd))
    {
        *events = EV_READ;
    }

    return rv;
}

void SocketSetNonblocking(int sock, int nb)
{
#if defined(_WIN32)
    unsigned int yes = nb;
    ioctlsocket(s, FIONBIO, &yes);
#else
    int flags = fcntl(sock, F_GETFL, 0);
    if (nb)
        flags |= O_NONBLOCK;
    else
        flags &= ~O_NONBLOCK;
    fcntl(sock, F_SETFL, flags);
#endif
}

int SocketGetOpt(int sock, int level, int name, void *val, int *len)
{
#if defined(_WIN32)
    return getsockopt(sock, level, name, (char *) val, len);
#else
    socklen_t _len = *len;
    int rc = getsockopt(sock, level, name, val, &_len);
    *len = _len;
    return rc;
#endif
}

int SocketGetError(int sock, int *error)
{
    int len = sizeof(*error);
    return SocketGetOpt(sock, SOL_SOCKET, SO_ERROR, error, &len);
}

int SocketWouldBlock(int error)
{
#if defined(_WIN32)
    return error == WSAEWOULDBLOCK;
#else
    return error == EWOULDBLOCK || error == EAGAIN;
#endif
}