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

**

** Copyright (C) 2010 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 "qhostaddress.h"

#include "qhostaddress_p.h"

#include "qdebug.h"

#include "qplatformdefs.h"

#include "qstringlist.h"

#include "qendian.h"

#ifndef QT_NO_DATASTREAM

#include <qdatastream.h>

#endif

#if defined(Q_OS_WINCE)

#include <winsock.h>

#endif

#ifdef QT_LINUXBASE

#  include <arpa/inet.h>

#endif

QT_BEGIN_NAMESPACE

#define QT_ENSURE_PARSED(a) \

    do { \

        if (!(a)->d->isParsed) \

            (a)->d->parse(); \

    } while (0)

#ifdef Q_OS_WIN

#    if !defined (QT_NO_IPV6)

// sockaddr_in6 size changed between old and new SDK

// Only the new version is the correct one, so always

// use this structure.

#if defined(Q_OS_WINCE)

#  if !defined(u_char)

#    define u_char unsigned char

#  endif

#  if !defined(u_short)

#    define u_short unsigned short

#  endif

#  if !defined(u_long)

#    define u_long unsigned long

#  endif

#endif

struct qt_in6_addr {

    u_char qt_s6_addr[16];

};

typedef struct {

    short   sin6_family;            /* AF_INET6 */

    u_short sin6_port;              /* Transport level port number */

    u_long  sin6_flowinfo;          /* IPv6 flow information */

    struct  qt_in6_addr sin6_addr;  /* IPv6 address */

    u_long  sin6_scope_id;          /* set of interfaces for a scope */

} qt_sockaddr_in6;

#    else

typedef void * qt_sockaddr_in6 ;

#    endif

#    ifndef AF_INET6

#        define AF_INET6        23  /* Internetwork Version 6 */

#    endif

#else

#define qt_sockaddr_in6 sockaddr_in6

#define qt_s6_addr s6_addr

#endif

class QHostAddressPrivate

{

public:

    QHostAddressPrivate();

    void setAddress(quint32 a_ = 0);

    void setAddress(const quint8 *a_);

    void setAddress(const Q_IPV6ADDR &a_);

    bool parse();

    void clear();

    quint32 a;    // IPv4 address

    Q_IPV6ADDR a6; // IPv6 address

    QAbstractSocket::NetworkLayerProtocol protocol;

    QString ipString;

    bool isParsed;

    QString scopeId;

    friend class QHostAddress;

};

QHostAddressPrivate::QHostAddressPrivate()

    : a(0), protocol(QAbstractSocket::UnknownNetworkLayerProtocol), isParsed(true)

{

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

}

void QHostAddressPrivate::setAddress(quint32 a_)

{

    a = a_;

    protocol = QAbstractSocket::IPv4Protocol;

    isParsed = true;

}

void QHostAddressPrivate::setAddress(const quint8 *a_)

{

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

        a6[i] = a_[i];

    protocol = QAbstractSocket::IPv6Protocol;

    isParsed = true;

}

void QHostAddressPrivate::setAddress(const Q_IPV6ADDR &a_)

{

    a6 = a_;

    a = 0;

    protocol = QAbstractSocket::IPv6Protocol;

    isParsed = true;

}

static bool parseIp4(const QString& address, quint32 *addr)

{

    QStringList ipv4 = address.split(QLatin1String("."));

    if (ipv4.count() != 4)

        return false;

    quint32 ipv4Address = 0;

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

        bool ok = false;

        uint byteValue = ipv4.at(i).toUInt(&ok);

        if (!ok || byteValue > 255)

            return false;

        ipv4Address <<= 8;

        ipv4Address += byteValue;

    }

    *addr = ipv4Address;

    return true;

}

static bool parseIp6(const QString &address, quint8 *addr, QString *scopeId)

{

    QString tmp = address;

    int scopeIdPos = tmp.lastIndexOf(QLatin1Char('%'));

    if (scopeIdPos != -1) {

        *scopeId = tmp.mid(scopeIdPos + 1);

        tmp.chop(tmp.size() - scopeIdPos);

    } else {

        scopeId->clear();

    }

    QStringList ipv6 = tmp.split(QLatin1String(":"));

    int count = ipv6.count();

    if (count < 3 || count > 8)

        return false;

    int colonColon = tmp.count(QLatin1String("::"));

    if(count == 8 && colonColon > 1)

        return false;

    // address can be compressed with a "::", but that

    // may only appear once (see RFC 1884)

    // the statement below means:

    // if(shortened notation is not used AND

    // ((pure IPv6 notation AND less than 8 parts) OR

    // ((mixed IPv4/6 notation AND less than 7 parts)))

    if(colonColon != 1 && count < (tmp.contains(QLatin1Char('.')) ? 7 : 8))

        return false;

    int mc = 16;

    int fillCount = 9 - count;  // number of 0 words to fill in the middle

    for (int i = count - 1; i >= 0; --i) {

        if (mc <= 0)

            return false;

        if (ipv6.at(i).isEmpty()) {

            if (i == count - 1) {

                // special case: ":" is last character

                if (!ipv6.at(i - 1).isEmpty())

                    return false;

                addr[--mc] = 0;

                addr[--mc] = 0;

            } else if (i == 0) {

                // special case: ":" is first character

                if (!ipv6.at(i + 1).isEmpty())

                    return false;

                addr[--mc] = 0;

                addr[--mc] = 0;

            } else {

                for (int j = 0; j < fillCount; ++j) {

                    if (mc <= 0)

                        return false;

                    addr[--mc] = 0;

                    addr[--mc] = 0;

                }

            }

        } else {

            bool ok = false;

            uint byteValue = ipv6.at(i).toUInt(&ok, 16);

            if (ok && byteValue <= 0xffff) {

                addr[--mc] = byteValue & 0xff;

                addr[--mc] = (byteValue >> 8) & 0xff;

            } else {

                if (i != count - 1)

                    return false;

                // parse the ipv4 part of a mixed type

                quint32 maybeIp4;

                if (!parseIp4(ipv6.at(i), &maybeIp4))

                    return false;

                addr[--mc] = maybeIp4 & 0xff;

                addr[--mc] = (maybeIp4 >> 8) & 0xff;

                addr[--mc] = (maybeIp4 >> 16) & 0xff;

                addr[--mc] = (maybeIp4 >> 24) & 0xff;

                --fillCount;

            }

        }

    }

    return true;

}

bool QHostAddressPrivate::parse()

{

    isParsed = true;

    protocol = QAbstractSocket::UnknownNetworkLayerProtocol;

    QString a = ipString.simplified();

    // All IPv6 addresses contain a ':', and may contain a '.'.

    if (a.contains(QLatin1Char(':'))) {

        quint8 maybeIp6[16];

        if (parseIp6(a, maybeIp6, &scopeId)) {

            setAddress(maybeIp6);

            protocol = QAbstractSocket::IPv6Protocol;

            return true;

        }

    }

    // All IPv4 addresses contain a '.'.

    if (a.contains(QLatin1Char('.'))) {

        quint32 maybeIp4 = 0;

        if (parseIp4(a, &maybeIp4)) {

            setAddress(maybeIp4);

            protocol = QAbstractSocket::IPv4Protocol;

            return true;

        }

    }

    return false;

}

void QHostAddressPrivate::clear()

{

    a = 0;

    protocol = QAbstractSocket::UnknownNetworkLayerProtocol;

    isParsed = true;

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

}

bool QNetmaskAddress::setAddress(const QString &address)

{

    length = -1;

    QHostAddress other;

    return other.setAddress(address) && setAddress(other);

}

bool QNetmaskAddress::setAddress(const QHostAddress &address)

{

    static const quint8 zeroes[16] = { 0 };

    union {

        quint32 v4;

        quint8 v6[16];

    } ip;

    int netmask = 0;

    quint8 *ptr = ip.v6;

    quint8 *end;

    length = -1;

    QHostAddress::operator=(address);

    if (d->protocol == QAbstractSocket::IPv4Protocol) {

        ip.v4 = qToBigEndian(d->a);

        end = ptr + 4;

    } else if (d->protocol == QAbstractSocket::IPv6Protocol) {

        memcpy(ip.v6, d->a6.c, 16);

        end = ptr + 16;

    } else {

        d->clear();

        return false;

    }

    while (ptr < end) {

        switch (*ptr) {

        case 255:

            netmask += 8;

            ++ptr;

            continue;

        default:

            d->clear();

            return false;       // invalid IP-style netmask

            // the rest always falls through

        case 254:

            ++netmask;

        case 252:

            ++netmask;

        case 248:

            ++netmask;

        case 240:

            ++netmask;

        case 224:

            ++netmask;

        case 192:

            ++netmask;

        case 128:

            ++netmask;

        case 0:

            break;

        }

        break;

    }

    // confirm that the rest is only zeroes

    if (ptr < end && memcmp(ptr + 1, zeroes, end - ptr - 1) != 0) {

        d->clear();

        return false;

    }

    length = netmask;

    return true;

}

static void clearBits(quint8 *where, int start, int end)

{

    Q_ASSERT(end == 32 || end == 128);

    if (start == end)

        return;

    // for the byte where 'start' is, clear the lower bits only

    quint8 bytemask = 256 - (1 << (8 - (start & 7)));

    where[start / 8] &= bytemask;

    // for the tail part, clear everything

    memset(where + (start + 7) / 8, 0, end / 8 - (start + 7) / 8);

}

int QNetmaskAddress::prefixLength() const

{

    return length;

}

void QNetmaskAddress::setPrefixLength(QAbstractSocket::NetworkLayerProtocol proto, int newLength)

{

    length = newLength;

    if (length < 0 || length > (proto == QAbstractSocket::IPv4Protocol ? 32 :

                                proto == QAbstractSocket::IPv6Protocol ? 128 : -1)) {

        // invalid information, reject

        d->protocol = QAbstractSocket::UnknownNetworkLayerProtocol;

        length = -1;

        return;

    }

    d->protocol = proto;

    if (d->protocol == QAbstractSocket::IPv4Protocol) {

        if (length == 0) {

            d->a = 0;

        } else if (length == 32) {

            d->a = quint32(0xffffffff);

        } else {

            d->a = quint32(0xffffffff) >> (32 - length) << (32 - length);

        }

    } else {

        memset(d->a6.c, 0xFF, sizeof(d->a6));

        clearBits(d->a6.c, length, 128);

    }

}

/*!

    \class QHostAddress

    \brief The QHostAddress class provides an IP address.

    \ingroup network

    \inmodule QtNetwork

    This class holds an IPv4 or IPv6 address in a platform- and

    protocol-independent manner.

    QHostAddress is normally used with the QTcpSocket, QTcpServer,

    and QUdpSocket to connect to a host or to set up a server.

    A host address is set with setAddress(), checked for its type

    using isIPv4Address() or isIPv6Address(), and retrieved with

    toIPv4Address(), toIPv6Address(), or toString().

    \note Please note that QHostAddress does not do DNS lookups.

    QHostInfo is needed for that.

    The class also supports common predefined addresses: \l Null, \l

    LocalHost, \l LocalHostIPv6, \l Broadcast, and \l Any.

    \sa QHostInfo, QTcpSocket, QTcpServer, QUdpSocket

*/

/*! \enum QHostAddress::SpecialAddress

    \value Null The null address object. Equivalent to QHostAddress().

    \value LocalHost The IPv4 localhost address. Equivalent to QHostAddress("127.0.0.1").

    \value LocalHostIPv6 The IPv6 localhost address. Equivalent to QHostAddress("::1").

    \value Broadcast The IPv4 broadcast address. Equivalent to QHostAddress("255.255.255.255").

    \value Any The IPv4 any-address. Equivalent to QHostAddress("0.0.0.0").

    \value AnyIPv6 The IPv6 any-address. Equivalent to QHostAddress("::").

*/

/*!  Constructs a host address object with the IP address 0.0.0.0.

    \sa clear()

*/

QHostAddress::QHostAddress()

    : d(new QHostAddressPrivate)

{

}

/*!

    Constructs a host address object with the IPv4 address \a ip4Addr.

*/

QHostAddress::QHostAddress(quint32 ip4Addr)

    : d(new QHostAddressPrivate)

{

    setAddress(ip4Addr);

}

/*!

    Constructs a host address object with the IPv6 address \a ip6Addr.

    \a ip6Addr must be a 16-byte array in network byte order (big

    endian).

*/

QHostAddress::QHostAddress(quint8 *ip6Addr)

    : d(new QHostAddressPrivate)

{

    setAddress(ip6Addr);

}

/*!

    Constructs a host address object with the IPv6 address \a ip6Addr.

*/

QHostAddress::QHostAddress(const Q_IPV6ADDR &ip6Addr)

    : d(new QHostAddressPrivate)

{

    setAddress(ip6Addr);

}

/*!

    Constructs an IPv4 or IPv6 address based on the string \a address

    (e.g., "127.0.0.1").

    \sa setAddress()

*/

QHostAddress::QHostAddress(const QString &address)

    : d(new QHostAddressPrivate)

{

    d->ipString = address;

    d->isParsed = false;

}

/*!

    \fn QHostAddress::QHostAddress(const sockaddr *sockaddr)

    Constructs an IPv4 or IPv6 address using the address specified by

    the native structure \a sockaddr.

    \sa setAddress()

*/

QHostAddress::QHostAddress(const struct sockaddr *sockaddr)

    : d(new QHostAddressPrivate)

{

    if (sockaddr->sa_family == AF_INET)

        setAddress(htonl(((sockaddr_in *)sockaddr)->sin_addr.s_addr));

#ifndef QT_NO_IPV6

    else if (sockaddr->sa_family == AF_INET6)

        setAddress(((qt_sockaddr_in6 *)sockaddr)->sin6_addr.qt_s6_addr);

#endif

}

/*!

    Constructs a copy of the given \a address.

*/

QHostAddress::QHostAddress(const QHostAddress &address)