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

**

** 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 "qhttpnetworkconnection_p.h"

#include "qhttpnetworkconnectionchannel_p.h"

#include "private/qnoncontiguousbytedevice_p.h"

#include <private/qnetworkrequest_p.h>

#include <private/qobject_p.h>

#include <private/qauthenticator_p.h>

#include <qnetworkproxy.h>

#include <qauthenticator.h>

#include <qbuffer.h>

#include <qpair.h>

#include <qhttp.h>

#include <qdebug.h>

#ifndef QT_NO_HTTP

#ifndef QT_NO_OPENSSL

#    include <QtNetwork/qsslkey.h>

#    include <QtNetwork/qsslcipher.h>

#    include <QtNetwork/qsslconfiguration.h>

#endif

QT_BEGIN_NAMESPACE

#ifdef Q_OS_SYMBIAN

const int QHttpNetworkConnectionPrivate::defaultChannelCount = 3;

#else

const int QHttpNetworkConnectionPrivate::defaultChannelCount = 6;

#endif

// the maximum amount of requests that might be pipelined into a socket

// from what was suggested, 3 seems to be OK

const int QHttpNetworkConnectionPrivate::defaultPipelineLength = 3;

QHttpNetworkConnectionPrivate::QHttpNetworkConnectionPrivate(const QString &hostName, quint16 port, bool encrypt)

: hostName(hostName), port(port), encrypt(encrypt),

  channelCount(defaultChannelCount),

  pendingAuthSignal(false), pendingProxyAuthSignal(false)

#ifndef QT_NO_NETWORKPROXY

  , networkProxy(QNetworkProxy::NoProxy)

#endif

{

    channels = new QHttpNetworkConnectionChannel[channelCount];

}

QHttpNetworkConnectionPrivate::QHttpNetworkConnectionPrivate(quint16 channelCount, const QString &hostName, quint16 port, bool encrypt)

: hostName(hostName), port(port), encrypt(encrypt),

  channelCount(channelCount),

  pendingAuthSignal(false), pendingProxyAuthSignal(false)

#ifndef QT_NO_NETWORKPROXY

  , networkProxy(QNetworkProxy::NoProxy)

#endif

{

    channels = new QHttpNetworkConnectionChannel[channelCount];

}

QHttpNetworkConnectionPrivate::~QHttpNetworkConnectionPrivate()

{

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

        if (channels[i].socket) {

            channels[i].socket->close();

            delete channels[i].socket;

        }

    }

    delete []channels;

}

void QHttpNetworkConnectionPrivate::init()

{

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

        channels[i].setConnection(this->q_func());

        channels[i].init();

    }

}

int QHttpNetworkConnectionPrivate::indexOf(QAbstractSocket *socket) const

{

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

        if (channels[i].socket == socket)

            return i;

    qFatal("Called with unknown socket object.");

    return 0;

}

qint64 QHttpNetworkConnectionPrivate::uncompressedBytesAvailable(const QHttpNetworkReply &reply) const

{

    return reply.d_func()->responseData.byteAmount();

}

qint64 QHttpNetworkConnectionPrivate::uncompressedBytesAvailableNextBlock(const QHttpNetworkReply &reply) const

{

    return reply.d_func()->responseData.sizeNextBlock();

}

void QHttpNetworkConnectionPrivate::prepareRequest(HttpMessagePair &messagePair)

{

    QHttpNetworkRequest &request = messagePair.first;

    QHttpNetworkReply *reply = messagePair.second;

    // add missing fields for the request

    QByteArray value;

    // check if Content-Length is provided

    QNonContiguousByteDevice* uploadByteDevice = request.uploadByteDevice();

    if (uploadByteDevice) {

        if (request.contentLength() != -1 && uploadByteDevice->size() != -1) {

            // both values known, take the smaller one.

            request.setContentLength(qMin(uploadByteDevice->size(), request.contentLength()));

        } else if (request.contentLength() == -1 && uploadByteDevice->size() != -1) {

            // content length not supplied by user, but the upload device knows it

            request.setContentLength(uploadByteDevice->size());

        } else if (request.contentLength() != -1 && uploadByteDevice->size() == -1) {

            // everything OK, the user supplied us the contentLength

        } else if (request.contentLength() == -1 && uploadByteDevice->size() == -1) {

            qFatal("QHttpNetworkConnectionPrivate: Neither content-length nor upload device size were given");

        }

    }

    // set the Connection/Proxy-Connection: Keep-Alive headers

#ifndef QT_NO_NETWORKPROXY

    if (networkProxy.type() == QNetworkProxy::HttpCachingProxy)  {

        value = request.headerField("proxy-connection");

        if (value.isEmpty())

            request.setHeaderField("Proxy-Connection", "Keep-Alive");

    } else {

#endif

        value = request.headerField("connection");

        if (value.isEmpty())

            request.setHeaderField("Connection", "Keep-Alive");

#ifndef QT_NO_NETWORKPROXY

    }

#endif

    // If the request had a accept-encoding set, we better not mess

    // with it. If it was not set, we announce that we understand gzip

    // and remember this fact in request.d->autoDecompress so that

    // we can later decompress the HTTP reply if it has such an

    // encoding.

    value = request.headerField("accept-encoding");

    if (value.isEmpty()) {

#ifndef QT_NO_COMPRESS

        request.setHeaderField("Accept-Encoding", "gzip");

        request.d->autoDecompress = true;

#else

        // if zlib is not available set this to false always

        request.d->autoDecompress = false;

#endif

    }

    // some websites mandate an accept-language header and fail

    // if it is not sent. This is a problem with the website and

    // not with us, but we work around this by setting

    // one always.

    value = request.headerField("accept-language");

    if (value.isEmpty()) {

        QString systemLocale = QLocale::system().name().replace(QChar::fromAscii('_'),QChar::fromAscii('-'));

        QString acceptLanguage;

        if (systemLocale == QLatin1String("C"))

            acceptLanguage = QString::fromAscii("en,*");

        else if (systemLocale.startsWith(QLatin1String("en-")))

            acceptLanguage = QString::fromAscii("%1,*").arg(systemLocale);

        else

            acceptLanguage = QString::fromAscii("%1,en,*").arg(systemLocale);

        request.setHeaderField("Accept-Language", acceptLanguage.toAscii());

    }

    // set the User Agent

    value = request.headerField("user-agent");

    if (value.isEmpty())

        request.setHeaderField("User-Agent", "Mozilla/5.0");

    // set the host

    value = request.headerField("host");

    if (value.isEmpty()) {

        QByteArray host = QUrl::toAce(hostName);

        int port = request.url().port();

        if (port != -1) {

            host += ':';

            host += QByteArray::number(port);

        }

        request.setHeaderField("Host", host);

    }

    reply->d_func()->requestIsPrepared = true;

}

void QHttpNetworkConnectionPrivate::emitReplyError(QAbstractSocket *socket,

                                                   QHttpNetworkReply *reply,

                                                   QNetworkReply::NetworkError errorCode)

{

    Q_Q(QHttpNetworkConnection);

    if (socket && reply) {

        // this error matters only to this reply

        reply->d_func()->errorString = errorDetail(errorCode, socket);

        emit reply->finishedWithError(errorCode, reply->d_func()->errorString);

        int i = indexOf(socket);

        // remove the corrupt data if any

        reply->d_func()->eraseData();

        channels[i].close();

        // send the next request

        QMetaObject::invokeMethod(q, "_q_startNextRequest", Qt::QueuedConnection);

    }

}

void QHttpNetworkConnectionPrivate::copyCredentials(int fromChannel, QAuthenticator *auth, bool isProxy)

{

    Q_ASSERT(auth);

    // select another channel

    QAuthenticator* otherAuth = 0;

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

        if (i == fromChannel)

            continue;

        if (isProxy)

            otherAuth = &channels[i].proxyAuthenticator;

        else

            otherAuth = &channels[i].authenticator;

        // if the credentials are different, copy them

        if (otherAuth->user().compare(auth->user()))

            otherAuth->setUser(auth->user());

        if (otherAuth->password().compare(auth->password()))

            otherAuth->setPassword(auth->password());

    }

}

// handles the authentication for one channel and eventually re-starts the other channels

bool QHttpNetworkConnectionPrivate::handleAuthenticateChallenge(QAbstractSocket *socket, QHttpNetworkReply *reply,

                                                                bool isProxy, bool &resend)

{

    Q_ASSERT(socket);

    Q_ASSERT(reply);

    Q_Q(QHttpNetworkConnection);

    resend = false;

    //create the response header to be used with QAuthenticatorPrivate.

    QHttpResponseHeader responseHeader;

    QList<QPair<QByteArray, QByteArray> > fields = reply->header();

    QList<QPair<QByteArray, QByteArray> >::const_iterator it = fields.constBegin();

    while (it != fields.constEnd()) {

        responseHeader.addValue(QString::fromLatin1(it->first), QString::fromUtf8(it->second));

        it++;

    }

    //find out the type of authentication protocol requested.

    QAuthenticatorPrivate::Method authMethod = reply->d_func()->authenticationMethod(isProxy);

    if (authMethod != QAuthenticatorPrivate::None) {

        int i = indexOf(socket);

        //Use a single authenticator for all domains. ### change later to use domain/realm

        QAuthenticator* auth = 0;

        if (isProxy) {

            auth = &channels[i].proxyAuthenticator;

            channels[i].proxyAuthMehtod = authMethod;

        } else {

            auth = &channels[i].authenticator;

            channels[i].authMehtod = authMethod;

        }

        //proceed with the authentication.

        if (auth->isNull())

            auth->detach();

        QAuthenticatorPrivate *priv = QAuthenticatorPrivate::getPrivate(*auth);

        priv->parseHttpResponse(responseHeader, isProxy);

        if (priv->phase == QAuthenticatorPrivate::Done) {

            if ((isProxy && pendingProxyAuthSignal) ||(!isProxy && pendingAuthSignal)) {

                // drop the request

                reply->d_func()->eraseData();

                channels[i].close();

                channels[i].lastStatus = 0;

                channels[i].state =  QHttpNetworkConnectionChannel::Wait4AuthState;

                return false;

            }

            // cannot use this socket until the slot returns

            channels[i].state = QHttpNetworkConnectionChannel::WaitingState;

            socket->blockSignals(true);

            if (!isProxy) {

                pendingAuthSignal = true;

                emit q->authenticationRequired(reply->request(), auth, q);

                pendingAuthSignal = false;

#ifndef QT_NO_NETWORKPROXY

            } else {

                pendingProxyAuthSignal = true;

                emit q->proxyAuthenticationRequired(networkProxy, auth, q);

                pendingProxyAuthSignal = false;

#endif

            }

            socket->blockSignals(false);

            // socket free to use

            channels[i].state = QHttpNetworkConnectionChannel::IdleState;

            if (priv->phase != QAuthenticatorPrivate::Done) {

                // send any pending requests

                copyCredentials(i,  auth, isProxy);

                QMetaObject::invokeMethod(q, "_q_restartAuthPendingRequests", Qt::QueuedConnection);

            }

        }

        // changing values in QAuthenticator will reset the 'phase'

        if (priv->phase == QAuthenticatorPrivate::Done) {

            // authentication is cancelled, send the current contents to the user.

            emit channels[i].reply->headerChanged();

            emit channels[i].reply->readyRead();

            QNetworkReply::NetworkError errorCode =

                isProxy

                ? QNetworkReply::ProxyAuthenticationRequiredError

                : QNetworkReply::AuthenticationRequiredError;

            reply->d_func()->errorString = errorDetail(errorCode, socket);

            emit q->error(errorCode, reply->d_func()->errorString);

            emit channels[i].reply->finished();

            // ### at this point the reply could be deleted

            socket->close();

            // remove pending request on the other channels

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

                if (j != i && channels[j].state ==  QHttpNetworkConnectionChannel::Wait4AuthState)

                    channels[j].state = QHttpNetworkConnectionChannel::IdleState;

            }

            return true;

        }

        //resend the request

        resend = true;

        return true;

    }

    return false;

}

void QHttpNetworkConnectionPrivate::createAuthorization(QAbstractSocket *socket, QHttpNetworkRequest &request)

{

    Q_ASSERT(socket);

    int i = indexOf(socket);

    if (channels[i].authMehtod != QAuthenticatorPrivate::None) {

        if (!(channels[i].authMehtod == QAuthenticatorPrivate::Ntlm && channels[i].lastStatus != 401)) {

            QAuthenticatorPrivate *priv = QAuthenticatorPrivate::getPrivate(channels[i].authenticator);

            if (priv && priv->method != QAuthenticatorPrivate::None) {

                QByteArray response = priv->calculateResponse(request.d->methodName(), request.d->uri(false));

                request.setHeaderField("Authorization", response);

            }

        }

    }

    if (channels[i].proxyAuthMehtod != QAuthenticatorPrivate::None) {

        if (!(channels[i].proxyAuthMehtod == QAuthenticatorPrivate::Ntlm && channels[i].lastStatus != 407)) {

            QAuthenticatorPrivate *priv = QAuthenticatorPrivate::getPrivate(channels[i].proxyAuthenticator);

            if (priv && priv->method != QAuthenticatorPrivate::None) {

                QByteArray response = priv->calculateResponse(request.d->methodName(), request.d->uri(false));

                request.setHeaderField("Proxy-Authorization", response);

            }

        }

    }

}

QHttpNetworkReply* QHttpNetworkConnectionPrivate::queueRequest(const QHttpNetworkRequest &request)

{

    Q_Q(QHttpNetworkConnection);

    // The reply component of the pair is created initially.

    QHttpNetworkReply *reply = new QHttpNetworkReply(request.url());

    reply->setRequest(request);

    reply->d_func()->connection = q;

    HttpMessagePair pair = qMakePair(request, reply);

    switch (request.priority()) {

    case QHttpNetworkRequest::HighPriority:

        highPriorityQueue.prepend(pair);

        break;

    case QHttpNetworkRequest::NormalPriority:

    case QHttpNetworkRequest::LowPriority:

        lowPriorityQueue.prepend(pair);

        break;

    }

    // this used to be called via invokeMethod and a QueuedConnection

    _q_startNextRequest();

    return reply;

}

void QHttpNetworkConnectionPrivate::requeueRequest(const HttpMessagePair &pair)

{

    QHttpNetworkRequest request = pair.first;

    switch (request.priority()) {

    case QHttpNetworkRequest::HighPriority:

        highPriorityQueue.prepend(pair);

        break;

    case QHttpNetworkRequest::NormalPriority:

    case QHttpNetworkRequest::LowPriority:

        lowPriorityQueue.prepend(pair);

        break;

    }

    // this used to be called via invokeMethod and a QueuedConnection

    _q_startNextRequest();

}

void QHttpNetworkConnectionPrivate::dequeueAndSendRequest(QAbstractSocket *socket)

{

    Q_ASSERT(socket);

    int i = indexOf(socket);

    if (!highPriorityQueue.isEmpty()) {

        // remove from queue before sendRequest! else we might pipeline the same request again

        HttpMessagePair messagePair = highPriorityQueue.takeLast();

        if (!messagePair.second->d_func()->requestIsPrepared)

            prepareRequest(messagePair);

        channels[i].request = messagePair.first;

        channels[i].reply = messagePair.second;

        channels[i].sendRequest();

        return;

    }

    if (!lowPriorityQueue.isEmpty()) {

        // remove from queue before sendRequest! else we might pipeline the same request again

        HttpMessagePair messagePair = lowPriorityQueue.takeLast();

        if (!messagePair.second->d_func()->requestIsPrepared)

            prepareRequest(messagePair);

        channels[i].request = messagePair.first;

        channels[i].reply = messagePair.second;

        channels[i].sendRequest();

        return;

    }

}

// this is called from _q_startNextRequest and when a request has been sent down a socket from the channel

void QHttpNetworkConnectionPrivate::fillPipeline(QAbstractSocket *socket)

{

    // return fast if there is nothing to pipeline

    if (highPriorityQueue.isEmpty() && lowPriorityQueue.isEmpty())

        return;

    int i = indexOf(socket);

    bool highPriorityQueueProcessingDone = false;

    bool lowPriorityQueueProcessingDone = false;

    while (!highPriorityQueueProcessingDone && !lowPriorityQueueProcessingDone) {

        // this loop runs once per request we intend to pipeline in.

        if (channels[i].pipeliningSupported != QHttpNetworkConnectionChannel::PipeliningProbablySupported)

            return;

        // the current request that is in must already support pipelining

        if (!channels[i].request.isPipeliningAllowed())

            return;

        // the current request must be a idempotent (right now we only check GET)

        if (channels[i].request.operation() != QHttpNetworkRequest::Get)

            return;

        // check if socket is connected

        if (socket->state() != QAbstractSocket::ConnectedState)

            return;

        // check for resendCurrent

        if (channels[i].resendCurrent)

            return;

        // we do not like authentication stuff

        // ### make sure to be OK with this in later releases

        if (!channels[i].authenticator.isNull() || !channels[i].authenticator.user().isEmpty())

            return;

        if (!channels[i].proxyAuthenticator.isNull() || !channels[i].proxyAuthenticator.user().isEmpty())

            return;

        // check for pipeline length

        if (channels[i].alreadyPipelinedRequests.length() >= defaultPipelineLength)

            return;

        // must be in ReadingState or WaitingState

        if (! (channels[i].state == QHttpNetworkConnectionChannel::WaitingState

               || channels[i].state == QHttpNetworkConnectionChannel::ReadingState))

            return;

        highPriorityQueueProcessingDone = fillPipeline(highPriorityQueue, channels[i]);

        // not finished with highPriorityQueue? then loop again

        if (!highPriorityQueueProcessingDone)

            continue;

        // highPriorityQueue was processed, now deal with the lowPriorityQueue

        lowPriorityQueueProcessingDone = fillPipeline(lowPriorityQueue, channels[i]);

    }

}