/****************************************************************************

**

** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtNetwork module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <winsock2.h>

#include "qnativesocketengine_p.h"

#include <qabstracteventdispatcher.h>

#include <qsocketnotifier.h>

#include <qdebug.h>

#include <qdatetime.h>

//#define QNATIVESOCKETENGINE_DEBUG

#if defined(QNATIVESOCKETENGINE_DEBUG)

#   include <qstring.h>

#   include <qbytearray.h>

#endif

QT_BEGIN_NAMESPACE

#if defined(QNATIVESOCKETENGINE_DEBUG)

void verboseWSErrorDebug(int r)

{

    switch (r) {

        case WSANOTINITIALISED : qDebug("WSA error : WSANOTINITIALISED"); break;

        case WSAEINTR: qDebug("WSA error : WSAEINTR"); break;

        case WSAEBADF: qDebug("WSA error : WSAEBADF"); break;

        case WSAEACCES: qDebug("WSA error : WSAEACCES"); break;

        case WSAEFAULT: qDebug("WSA error : WSAEFAULT"); break;

        case WSAEINVAL: qDebug("WSA error : WSAEINVAL"); break;

        case WSAEMFILE: qDebug("WSA error : WSAEMFILE"); break;

        case WSAEWOULDBLOCK: qDebug("WSA error : WSAEWOULDBLOCK"); break;

        case WSAEINPROGRESS: qDebug("WSA error : WSAEINPROGRESS"); break;

        case WSAEALREADY: qDebug("WSA error : WSAEALREADY"); break;

        case WSAENOTSOCK: qDebug("WSA error : WSAENOTSOCK"); break;

        case WSAEDESTADDRREQ: qDebug("WSA error : WSAEDESTADDRREQ"); break;

        case WSAEMSGSIZE: qDebug("WSA error : WSAEMSGSIZE"); break;

        case WSAEPROTOTYPE: qDebug("WSA error : WSAEPROTOTYPE"); break;

        case WSAENOPROTOOPT: qDebug("WSA error : WSAENOPROTOOPT"); break;

        case WSAEPROTONOSUPPORT: qDebug("WSA error : WSAEPROTONOSUPPORT"); break;

        case WSAESOCKTNOSUPPORT: qDebug("WSA error : WSAESOCKTNOSUPPORT"); break;

        case WSAEOPNOTSUPP: qDebug("WSA error : WSAEOPNOTSUPP"); break;

        case WSAEPFNOSUPPORT: qDebug("WSA error : WSAEPFNOSUPPORT"); break;

        case WSAEAFNOSUPPORT: qDebug("WSA error : WSAEAFNOSUPPORT"); break;

        case WSAEADDRINUSE: qDebug("WSA error : WSAEADDRINUSE"); break;

        case WSAEADDRNOTAVAIL: qDebug("WSA error : WSAEADDRNOTAVAIL"); break;

        case WSAENETDOWN: qDebug("WSA error : WSAENETDOWN"); break;

        case WSAENETUNREACH: qDebug("WSA error : WSAENETUNREACH"); break;

        case WSAENETRESET: qDebug("WSA error : WSAENETRESET"); break;

        case WSAECONNABORTED: qDebug("WSA error : WSAECONNABORTED"); break;

        case WSAECONNRESET: qDebug("WSA error : WSAECONNRESET"); break;

        case WSAENOBUFS: qDebug("WSA error : WSAENOBUFS"); break;

        case WSAEISCONN: qDebug("WSA error : WSAEISCONN"); break;

        case WSAENOTCONN: qDebug("WSA error : WSAENOTCONN"); break;

        case WSAESHUTDOWN: qDebug("WSA error : WSAESHUTDOWN"); break;

        case WSAETOOMANYREFS: qDebug("WSA error : WSAETOOMANYREFS"); break;

        case WSAETIMEDOUT: qDebug("WSA error : WSAETIMEDOUT"); break;

        case WSAECONNREFUSED: qDebug("WSA error : WSAECONNREFUSED"); break;

        case WSAELOOP: qDebug("WSA error : WSAELOOP"); break;

        case WSAENAMETOOLONG: qDebug("WSA error : WSAENAMETOOLONG"); break;

        case WSAEHOSTDOWN: qDebug("WSA error : WSAEHOSTDOWN"); break;

        case WSAEHOSTUNREACH: qDebug("WSA error : WSAEHOSTUNREACH"); break;

        case WSAENOTEMPTY: qDebug("WSA error : WSAENOTEMPTY"); break;

        case WSAEPROCLIM: qDebug("WSA error : WSAEPROCLIM"); break;

        case WSAEUSERS: qDebug("WSA error : WSAEUSERS"); break;

        case WSAEDQUOT: qDebug("WSA error : WSAEDQUOT"); break;

        case WSAESTALE: qDebug("WSA error : WSAESTALE"); break;

        case WSAEREMOTE: qDebug("WSA error : WSAEREMOTE"); break;

        case WSAEDISCON: qDebug("WSA error : WSAEDISCON"); break;

        default: qDebug("WSA error : Unknown"); break;

    }

    qErrnoWarning(r, "more details");

}

/*

    Returns a human readable representation of the first \a len

    characters in \a data.

*/

static QByteArray qt_prettyDebug(const char *data, int len, int maxLength)

{

    if (!data) return "(null)";

    QByteArray out;

    for (int i = 0; i < len; ++i) {

        char c = data[i];

        if (isprint(int(uchar(c)))) {

            out += c;

        } else switch (c) {

        case '\n': out += "\\n"; break;

        case '\r': out += "\\r"; break;

        case '\t': out += "\\t"; break;

        default:

            QString tmp;

            tmp.sprintf("\\%o", c);

            out += tmp.toLatin1().constData();

        }

    }

    if (len < maxLength)

        out += "...";

    return out;

}

#define WS_ERROR_DEBUG(x) verboseWSErrorDebug(x);

#else

#define WS_ERROR_DEBUG(x) Q_UNUSED(x)

#endif

#ifndef AF_INET6

#define AF_INET6        23              /* Internetwork Version 6 */

#endif

#ifndef SO_EXCLUSIVEADDRUSE

#define SO_EXCLUSIVEADDRUSE ((int)(~SO_REUSEADDR)) /* disallow local address reuse */

#endif

//###

#define QT_SOCKLEN_T int

#define QT_SOCKOPTLEN_T int

/*

    Extracts the port and address from a sockaddr, and stores them in

    \a port and \a addr if they are non-null.

*/

static inline void qt_socket_getPortAndAddress(SOCKET socketDescriptor, const qt_sockaddr *sa, quint16 *port, QHostAddress *address)

{

#if !defined (QT_NO_IPV6)

    if (sa->a.sa_family == AF_INET6) {

        const qt_sockaddr_in6 *sa6 = &sa->a6;

        Q_IPV6ADDR tmp;

        for (int i = 0; i < 16; ++i)

            tmp.c[i] = sa6->sin6_addr.qt_s6_addr[i];

        QHostAddress a;

	a.setAddress(tmp);

	if (address)

	    *address = a;

        if (port)

	    WSANtohs(socketDescriptor, sa6->sin6_port, port);

    } else

#endif

    if (sa->a.sa_family == AF_INET) {

        const sockaddr_in *sa4 = &sa->a4;

        unsigned long addr;

        WSANtohl(socketDescriptor, sa4->sin_addr.s_addr, &addr);

        QHostAddress a;

	a.setAddress(addr);

	if (address)

	    *address = a;

        if (port)

	    WSANtohs(socketDescriptor, sa4->sin_port, port);

    }

}

/*! \internal

    Sets the port and address to a sockaddr. Requires that sa point to the IPv6 struct if the address is IPv6.

*/

static inline void qt_socket_setPortAndAddress(SOCKET socketDescriptor, sockaddr_in * sockAddrIPv4, qt_sockaddr_in6 * sockAddrIPv6,

                                               quint16 port, const QHostAddress & address, sockaddr ** sockAddrPtr, QT_SOCKLEN_T *sockAddrSize)

{

#if !defined(QT_NO_IPV6)

    if (address.protocol() == QAbstractSocket::IPv6Protocol) {

        memset(sockAddrIPv6, 0, sizeof(qt_sockaddr_in6));

        sockAddrIPv6->sin6_family = AF_INET6;

        WSAHtons(socketDescriptor, port, &(sockAddrIPv6->sin6_port));

        Q_IPV6ADDR tmp = address.toIPv6Address();

        memcpy(&(sockAddrIPv6->sin6_addr.qt_s6_addr), &tmp, sizeof(tmp));

        *sockAddrSize = sizeof(qt_sockaddr_in6);

        *sockAddrPtr = (struct sockaddr *) sockAddrIPv6;

    } else

#endif

    if (address.protocol() == QAbstractSocket::IPv4Protocol

        || address.protocol() == QAbstractSocket::UnknownNetworkLayerProtocol) {

        memset(sockAddrIPv4, 0, sizeof(sockaddr_in));

        sockAddrIPv4->sin_family = AF_INET;

        WSAHtons(socketDescriptor, port, &(sockAddrIPv4->sin_port));

        WSAHtonl(socketDescriptor, address.toIPv4Address(), &(sockAddrIPv4->sin_addr.s_addr));

        *sockAddrSize = sizeof(sockaddr_in);

        *sockAddrPtr = (struct sockaddr *) sockAddrIPv4;

    } else {

        // unreachable

    }

}

/*! \internal

*/

static inline QAbstractSocket::SocketType qt_socket_getType(int socketDescriptor)

{

    int value = 0;

    QT_SOCKLEN_T valueSize = sizeof(value);

    if (::getsockopt(socketDescriptor, SOL_SOCKET, SO_TYPE, (char *) &value, &valueSize) != 0) {

        WS_ERROR_DEBUG(WSAGetLastError());

    } else {

        if (value == SOCK_STREAM)

            return QAbstractSocket::TcpSocket;

        else if (value == SOCK_DGRAM)

            return QAbstractSocket::UdpSocket;

    }

    return QAbstractSocket::UnknownSocketType;

}

/*! \internal

*/

static inline int qt_socket_getMaxMsgSize(int socketDescriptor)

{

    int value = 0;

    QT_SOCKLEN_T valueSize = sizeof(value);

    if (::getsockopt(socketDescriptor, SOL_SOCKET, SO_MAX_MSG_SIZE, (char *) &value, &valueSize) != 0) {

        WS_ERROR_DEBUG(WSAGetLastError());

    }

    return value;

}

QWindowsSockInit::QWindowsSockInit()

:   version(0)

{

    //### should we try for 2.2 on all platforms ??

    WSAData wsadata;

    // IPv6 requires Winsock v2.0 or better.

    if (WSAStartup(MAKEWORD(2,0), &wsadata) != 0) {

	qWarning("QTcpSocketAPI: WinSock v2.0 initialization failed.");

    } else {

        version = 0x20;

    }

}

QWindowsSockInit::~QWindowsSockInit()

{

    WSACleanup();

}

// MS Transport Provider IOCTL to control

// reporting PORT_UNREACHABLE messages

// on UDP sockets via recv/WSARecv/etc.

// Path TRUE in input buffer to enable (default if supported),

// FALSE to disable.

#ifndef SIO_UDP_CONNRESET

#  ifndef IOC_VENDOR

#    define IOC_VENDOR 0x18000000

#  endif

#  ifndef _WSAIOW

#    define _WSAIOW(x,y) (IOC_IN|(x)|(y))

#  endif

#  define SIO_UDP_CONNRESET _WSAIOW(IOC_VENDOR,12)

#endif

bool QNativeSocketEnginePrivate::createNewSocket(QAbstractSocket::SocketType socketType, QAbstractSocket::NetworkLayerProtocol socketProtocol)

{

    //### no ip6 support on winsocket 1.1 but we will try not to use this !!!!!!!!!!!!1

    /*

    if (winsockVersion < 0x20 && socketProtocol == QAbstractSocket::IPv6Protocol) {

        //### no ip6 support

        return -1;

    }

    */

    int protocol = (socketProtocol == QAbstractSocket::IPv6Protocol) ? AF_INET6 : AF_INET;

    int type = (socketType == QAbstractSocket::UdpSocket) ? SOCK_DGRAM : SOCK_STREAM;

    // MSDN KB179942 states that on winnt 4 WSA_FLAG_OVERLAPPED is needed if socket is to be non blocking

    // and recomends alwasy doing it for cross windows version comapablity.

    SOCKET socket = ::WSASocket(protocol, type, 0, NULL, 0, WSA_FLAG_OVERLAPPED);

    if (socket == INVALID_SOCKET) {

        int err = WSAGetLastError();

        WS_ERROR_DEBUG(err);

        switch (err) {

        case WSANOTINITIALISED:

            //###

            break;

        case WSAEAFNOSUPPORT:

        case WSAESOCKTNOSUPPORT:

        case WSAEPROTOTYPE:

        case WSAEINVAL:

            setError(QAbstractSocket::UnsupportedSocketOperationError, ProtocolUnsupportedErrorString);

            break;

        case WSAEMFILE:

        case WSAENOBUFS:

            setError(QAbstractSocket::SocketResourceError, ResourceErrorString);

            break;

        default:

            break;

        }

        return false;

    }

#if !defined(Q_OS_WINCE)

    // enable new behavior using

    // SIO_UDP_CONNRESET

    DWORD dwBytesReturned = 0;

    int bNewBehavior = 1;

    if (::WSAIoctl(socket, SIO_UDP_CONNRESET, &bNewBehavior, sizeof(bNewBehavior),

                   NULL, 0, &dwBytesReturned, NULL, NULL) == SOCKET_ERROR) {

        // not to worry isBogusUdpReadNotification() should handle this otherwise

        int err = WSAGetLastError();

        WS_ERROR_DEBUG(err);

    }

#endif

    socketDescriptor = socket;

    return true;

}

/*! \internal

    Returns the value of the socket option \a opt.

*/

int QNativeSocketEnginePrivate::option(QNativeSocketEngine::SocketOption opt) const

{

    Q_Q(const QNativeSocketEngine);

    if (!q->isValid())

        return -1;

    int n = -1;

    int level = SOL_SOCKET; // default

    switch (opt) {

    case QNativeSocketEngine::ReceiveBufferSocketOption:

        n = SO_RCVBUF;

        break;

    case QNativeSocketEngine::SendBufferSocketOption:

        n = SO_SNDBUF;

        break;

    case QNativeSocketEngine::BroadcastSocketOption:

        n = SO_BROADCAST;

        break;

    case QNativeSocketEngine::NonBlockingSocketOption: {

        unsigned long buf = 0;

        if (WSAIoctl(socketDescriptor, FIONBIO, 0,0, &buf, sizeof(buf), 0,0,0) == 0)

            return buf;

        else

            return -1;

        break;

    }

    case QNativeSocketEngine::AddressReusable:

        n = SO_REUSEADDR;

        break;

    case QNativeSocketEngine::BindExclusively:

        n = SO_EXCLUSIVEADDRUSE;

        break;

    case QNativeSocketEngine::ReceiveOutOfBandData:

        n = SO_OOBINLINE;

        break;

    case QNativeSocketEngine::LowDelayOption:

        level = IPPROTO_TCP;

        n = TCP_NODELAY;

        break;

    case QNativeSocketEngine::KeepAliveOption:

        n = SO_KEEPALIVE;

        break;

    }

    int v = -1;

    QT_SOCKOPTLEN_T len = sizeof(v);

    if (getsockopt(socketDescriptor, level, n, (char *) &v, &len) != -1)

        return v;

    return -1;

}

/*! \internal

    Sets the socket option \a opt to \a v.

*/

bool QNativeSocketEnginePrivate::setOption(QNativeSocketEngine::SocketOption opt, int v)

{

    Q_Q(const QNativeSocketEngine);

    if (!q->isValid())

        return false;

    int n = 0;

    int level = SOL_SOCKET; // default

    switch (opt) {

    case QNativeSocketEngine::ReceiveBufferSocketOption:

        n = SO_RCVBUF;

        break;

    case QNativeSocketEngine::SendBufferSocketOption:

        n = SO_SNDBUF;

        break;

    case QNativeSocketEngine::BroadcastSocketOption:

        n = SO_BROADCAST;

        break;

    case QNativeSocketEngine::NonBlockingSocketOption:

        {

        unsigned long buf = v;

        unsigned long outBuf;

        DWORD sizeWritten = 0;

        if (::WSAIoctl(socketDescriptor, FIONBIO, &buf, sizeof(unsigned long), &outBuf, sizeof(unsigned long), &sizeWritten, 0,0) == SOCKET_ERROR) {

            WS_ERROR_DEBUG(WSAGetLastError());

            return false;

        }

        return true;

        break;

        }

    case QNativeSocketEngine::AddressReusable:

        n = SO_REUSEADDR;

        break;

    case QNativeSocketEngine::BindExclusively:

        n = SO_EXCLUSIVEADDRUSE;

        break;

    case QNativeSocketEngine::ReceiveOutOfBandData:

        n = SO_OOBINLINE;

        break;

    case QNativeSocketEngine::LowDelayOption:

        level = IPPROTO_TCP;

        n = TCP_NODELAY;

        break;

    case QNativeSocketEngine::KeepAliveOption:

        n = SO_KEEPALIVE;

        break;

    }

    if (::setsockopt(socketDescriptor, level, n, (char*)&v, sizeof(v)) != 0) {

        WS_ERROR_DEBUG(WSAGetLastError());

        return false;

    }

    return true;

}

/*!

    Fetches information about both ends of the connection: whatever is

    available.

*/

bool QNativeSocketEnginePrivate::fetchConnectionParameters()

{

    localPort = 0;

    localAddress.clear();

    peerPort = 0;

    peerAddress.clear();

    if (socketDescriptor == -1)

       return false;

    qt_sockaddr sa;

    QT_SOCKLEN_T sockAddrSize = sizeof(sa);

    // Determine local address

    memset(&sa, 0, sizeof(sa));

    if (::getsockname(socketDescriptor, &sa.a, &sockAddrSize) == 0) {

        qt_socket_getPortAndAddress(socketDescriptor, &sa, &localPort, &localAddress);

        // Determine protocol family

        switch (sa.a.sa_family) {

        case AF_INET:

            socketProtocol = QAbstractSocket::IPv4Protocol;

            break;

#if !defined (QT_NO_IPV6)

        case AF_INET6:

            socketProtocol = QAbstractSocket::IPv6Protocol;

            break;

#endif

        default:

            socketProtocol = QAbstractSocket::UnknownNetworkLayerProtocol;

            break;

        }

    } else {

        int err = WSAGetLastError();

        WS_ERROR_DEBUG(err);

        if (err == WSAENOTSOCK) {

            setError(QAbstractSocket::UnsupportedSocketOperationError,

                InvalidSocketErrorString);

            return false;

        }

    }

    memset(&sa, 0, sizeof(sa));

    if (::getpeername(socketDescriptor, &sa.a, &sockAddrSize) == 0) {

        qt_socket_getPortAndAddress(socketDescriptor, &sa, &peerPort, &peerAddress);

    } else {

        WS_ERROR_DEBUG(WSAGetLastError());

    }

    socketType = qt_socket_getType(socketDescriptor);

#if defined (QNATIVESOCKETENGINE_DEBUG)

    QString socketProtocolStr = "UnknownProtocol";

    if (socketProtocol == QAbstractSocket::IPv4Protocol) socketProtocolStr = "IPv4Protocol";

    else if (socketProtocol == QAbstractSocket::IPv6Protocol) socketProtocolStr = "IPv6Protocol";