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

**

** 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 test suite 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 <QtTest/QtTest>

#include <qcoreapplication.h>

#include <private/qhttpsocketengine_p.h>

#include <qhostinfo.h>

#include <qhostaddress.h>

#include <qtcpsocket.h>

#include <qhttp.h>

#include <qdebug.h>

#include <qtcpserver.h>

#include "../network-settings.h"

class tst_QHttpSocketEngine : public QObject

{

    Q_OBJECT

public:

    tst_QHttpSocketEngine();

    virtual ~tst_QHttpSocketEngine();

public slots:

    void init();

    void cleanup();

private slots:

    void construction();

    void errorTest_data();

    void errorTest();

    void simpleConnectToIMAP();

    void simpleErrorsAndStates();

    void tcpSocketBlockingTest();

    void tcpSocketNonBlockingTest();

    void downloadBigFile();

   // void tcpLoopbackPerformance();

    void passwordAuth();

protected slots:

    void tcpSocketNonBlocking_hostFound();

    void tcpSocketNonBlocking_connected();

    void tcpSocketNonBlocking_closed();

    void tcpSocketNonBlocking_readyRead();

    void tcpSocketNonBlocking_bytesWritten(qint64);

    void exitLoopSlot();

    void downloadBigFileSlot();

private:

    QTcpSocket *tcpSocketNonBlocking_socket;

    QStringList tcpSocketNonBlocking_data;

    qint64 tcpSocketNonBlocking_totalWritten;

    QTcpSocket *tmpSocket;

    qint64 bytesAvailable;

};

class MiniHttpServer: public QTcpServer

{

    Q_OBJECT

    QTcpSocket *client;

    QList<QByteArray> dataToTransmit;

public:

    QByteArray receivedData;

    MiniHttpServer(const QList<QByteArray> &data) : client(0), dataToTransmit(data)

    {

        listen();

        connect(this, SIGNAL(newConnection()), this, SLOT(doAccept()));

    }

public slots:

    void doAccept()

    {

        client = nextPendingConnection();

        connect(client, SIGNAL(readyRead()), this, SLOT(sendData()));

    }

    void sendData()

    {

        receivedData += client->readAll();

        int idx = client->property("dataTransmitionIdx").toInt();

        if (receivedData.contains("\r\n\r\n") ||

            receivedData.contains("\n\n")) {

            if (idx < dataToTransmit.length())

                client->write(dataToTransmit.at(idx++));

            if (idx == dataToTransmit.length()) {

                client->disconnectFromHost();

                disconnect(client, 0, this, 0);

                client = 0;

            } else {

                client->setProperty("dataTransmitionIdx", idx);

            }

        }

    }

};

tst_QHttpSocketEngine::tst_QHttpSocketEngine()

{

    Q_SET_DEFAULT_IAP

}

tst_QHttpSocketEngine::~tst_QHttpSocketEngine()

{

}

void tst_QHttpSocketEngine::init()

{

    tmpSocket = 0;

    bytesAvailable = 0;

}

void tst_QHttpSocketEngine::cleanup()

{

}

//---------------------------------------------------------------------------

void tst_QHttpSocketEngine::construction()

{

    QHttpSocketEngine socketDevice;

    QVERIFY(!socketDevice.isValid());

    // Initialize device

    QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));

    QVERIFY(socketDevice.isValid());

    QVERIFY(socketDevice.protocol() == QAbstractSocket::IPv4Protocol);

    QVERIFY(socketDevice.socketType() == QAbstractSocket::TcpSocket);

    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);

   // QVERIFY(socketDevice.socketDescriptor() != -1);

    QVERIFY(socketDevice.localAddress() == QHostAddress());

    QVERIFY(socketDevice.localPort() == 0);

    QVERIFY(socketDevice.peerAddress() == QHostAddress());

    QVERIFY(socketDevice.peerPort() == 0);

    QVERIFY(socketDevice.error() == QAbstractSocket::UnknownSocketError);

    //QTest::ignoreMessage(QtWarningMsg, "QSocketLayer::bytesAvailable() was called in QAbstractSocket::UnconnectedState");

    QVERIFY(socketDevice.bytesAvailable() == 0);

    //QTest::ignoreMessage(QtWarningMsg, "QSocketLayer::hasPendingDatagrams() was called in QAbstractSocket::UnconnectedState");

    QVERIFY(!socketDevice.hasPendingDatagrams());

}

//---------------------------------------------------------------------------

void tst_QHttpSocketEngine::errorTest_data()

{

    QTest::addColumn<QString>("hostname");

    QTest::addColumn<int>("port");

    QTest::addColumn<QString>("username");

    QTest::addColumn<QString>("response");

    QTest::addColumn<int>("expectedError");

    QQueue<QByteArray> responses;

    QTest::newRow("proxy-host-not-found") << "this-host-does-not-exist." << 1080 << QString()

                                          << QString()

                                          << int(QAbstractSocket::ProxyNotFoundError);

    QTest::newRow("proxy-connection-refused") << QtNetworkSettings::serverName() << 2 << QString()

                                              << QString()

                                              << int(QAbstractSocket::ProxyConnectionRefusedError);

    QTest::newRow("garbage1") << QString() << 0 << QString()

                              << "This is not HTTP\r\n\r\n"

                              << int(QAbstractSocket::ProxyProtocolError);

    QTest::newRow("garbage2") << QString() << 0 << QString()

                              << "This is not HTTP"

                              << int(QAbstractSocket::ProxyConnectionClosedError);

    QTest::newRow("garbage3") << QString() << 0 << QString()

                              << ""

                              << int(QAbstractSocket::ProxyConnectionClosedError);

    QTest::newRow("forbidden") << QString() << 0 << QString()

                               << "HTTP/1.0 403 Forbidden\r\n\r\n"

                               << int(QAbstractSocket::SocketAccessError);

    QTest::newRow("method-not-allowed") << QString() << 0 << QString()

                                        << "HTTP/1.0 405 Method Not Allowed\r\n\r\n"

                                        << int(QAbstractSocket::SocketAccessError);

    QTest::newRow("proxy-authentication-too-short")

        << QString() << 0 << "foo"

        << "HTTP/1.0 407 Proxy Authentication Required\r\n\r\n"

        << int(QAbstractSocket::ProxyProtocolError);

    QTest::newRow("proxy-authentication-invalid-method")

        << QString() << 0 << "foo"

        << "HTTP/1.0 407 Proxy Authentication Required\r\n"

           "Proxy-Authenticate: Frobnicator\r\n\r\n"

        << int(QAbstractSocket::ProxyProtocolError);

    QTest::newRow("proxy-authentication-required")

        << QString() << 0 << "foo"

        << "HTTP/1.0 407 Proxy Authentication Required\r\n"

           "Proxy-Connection: close\r\n"

           "Proxy-Authenticate: Basic, realm=wonderland\r\n\r\n"

        << int(QAbstractSocket::ProxyAuthenticationRequiredError);

    QTest::newRow("proxy-authentication-required2")

        << QString() << 0 << "foo"

        << "HTTP/1.0 407 Proxy Authentication Required\r\n"

           "Proxy-Connection: keep-alive\r\n"

           "Proxy-Authenticate: Basic, realm=wonderland\r\n\r\n"

           "\1"

           "HTTP/1.0 407 Proxy Authentication Required\r\n"

           "Proxy-Authenticate: Basic, realm=wonderland\r\n\r\n"

        << int(QAbstractSocket::ProxyAuthenticationRequiredError);

    QTest::newRow("proxy-authentication-required-noclose")

        << QString() << 0 << "foo"

        << "HTTP/1.0 407 Proxy Authentication Required\r\n"

           "Proxy-Authenticate: Basic\r\n"

           "\r\n"

        << int(QAbstractSocket::ProxyAuthenticationRequiredError);

    QTest::newRow("connection-refused") << QString() << 0 << QString()

                                        << "HTTP/1.0 503 Service Unavailable\r\n\r\n"

                                        << int(QAbstractSocket::ConnectionRefusedError);

    QTest::newRow("host-not-found") << QString() << 0 << QString()

                                    << "HTTP/1.0 404 Not Found\r\n\r\n"

                                    << int(QAbstractSocket::HostNotFoundError);

    QTest::newRow("weird-http-reply") << QString() << 0 << QString()

                                      << "HTTP/1.0 206 Partial Content\r\n\r\n"

                                      << int(QAbstractSocket::ProxyProtocolError);

}

void tst_QHttpSocketEngine::errorTest()

{

    QFETCH(QString, hostname);

    QFETCH(int, port);

    QFETCH(QString, username);

    QFETCH(QString, response);

    QFETCH(int, expectedError);

    MiniHttpServer server(response.toLatin1().split('\1'));

    if (hostname.isEmpty()) {

        hostname = "127.0.0.1";

        port = server.serverPort();

    }

    QTcpSocket socket;

    socket.setProxy(QNetworkProxy(QNetworkProxy::HttpProxy, hostname, port, username, username));

    socket.connectToHost("0.1.2.3", 12345);

    connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)),

            &QTestEventLoop::instance(), SLOT(exitLoop()));

    QTestEventLoop::instance().enterLoop(5);

    QVERIFY(!QTestEventLoop::instance().timeout());

    QCOMPARE(int(socket.error()), expectedError);

}

//---------------------------------------------------------------------------

void tst_QHttpSocketEngine::simpleConnectToIMAP()

{

    QHttpSocketEngine socketDevice;

    // Initialize device

    QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));

    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);

    socketDevice.setProxy(QNetworkProxy(QNetworkProxy::HttpProxy, QtNetworkSettings::serverName(), 3128));

    QVERIFY(!socketDevice.connectToHost(QtNetworkSettings::serverIP(), 143));

    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectingState);

    QVERIFY(socketDevice.waitForWrite());

    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectedState);

    QVERIFY(socketDevice.peerAddress() == QtNetworkSettings::serverIP());

    QVERIFY(!socketDevice.localAddress().isNull());

    QVERIFY(socketDevice.localPort() > 0);

    // Wait for the greeting

    QVERIFY(socketDevice.waitForRead());

    // Read the greeting

    qint64 available = socketDevice.bytesAvailable();

    QVERIFY(available > 0);

    QByteArray array;

    array.resize(available);

    QVERIFY(socketDevice.read(array.data(), array.size()) == available);

    // Check that the greeting is what we expect it to be

    QCOMPARE(array.constData(), QtNetworkSettings::expectedReplyIMAP().constData());

    // Write a logout message

    QByteArray array2 = "XXXX LOGOUT\r\n";

    QVERIFY(socketDevice.write(array2.data(),

                              array2.size()) == array2.size());

    // Wait for the response

    QVERIFY(socketDevice.waitForRead());

    available = socketDevice.bytesAvailable();

    QVERIFY(available > 0);

    array.resize(available);

    QVERIFY(socketDevice.read(array.data(), array.size()) == available);

    // Check that the greeting is what we expect it to be

    QCOMPARE(array.constData(), "* BYE LOGOUT received\r\nXXXX OK Completed\r\n");

    // Wait for the response

    QVERIFY(socketDevice.waitForRead());

    char c;

    QCOMPARE(socketDevice.read(&c, sizeof(c)), (qint64) -1);

    QVERIFY(socketDevice.error() == QAbstractSocket::RemoteHostClosedError);

    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);

}

//---------------------------------------------------------------------------

void tst_QHttpSocketEngine::simpleErrorsAndStates()

{

    {

        QHttpSocketEngine socketDevice;

        // Initialize device

        QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));

        socketDevice.setProxy(QNetworkProxy(QNetworkProxy::HttpProxy, QtNetworkSettings::serverName(), 3128));

        QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);

        QVERIFY(!socketDevice.connectToHost(QHostAddress(QtNetworkSettings::serverName()), 8088));

        QVERIFY(socketDevice.state() == QAbstractSocket::ConnectingState);

        if (socketDevice.waitForWrite(15000)) {

            QVERIFY(socketDevice.state() == QAbstractSocket::ConnectedState ||

                    socketDevice.state() == QAbstractSocket::UnconnectedState);

        } else {

            QVERIFY(socketDevice.error() == QAbstractSocket::SocketTimeoutError);

        }

    }

}

/*

//---------------------------------------------------------------------------

void tst_QHttpSocketEngine::tcpLoopbackPerformance()

{

    QTcpServer server;

    // Bind to any port on all interfaces

    QVERIFY(server.bind(QHostAddress("0.0.0.0"), 0));

    QVERIFY(server.state() == QAbstractSocket::BoundState);

    quint16 port = server.localPort();

    // Listen for incoming connections

    QVERIFY(server.listen());

    QVERIFY(server.state() == QAbstractSocket::ListeningState);

    // Initialize a Tcp socket

    QHttpSocketEngine client;

    QVERIFY(client.initialize(QAbstractSocket::TcpSocket));

    client.setProxy(QHostAddress("80.232.37.158"), 1081);

    // Connect to our server

    if (!client.connectToHost(QHostAddress("127.0.0.1"), port)) {

        QVERIFY(client.waitForWrite());

        QVERIFY(client.connectToHost(QHostAddress("127.0.0.1"), port));

    }

    // The server accepts the connectio

    int socketDescriptor = server.accept();

    QVERIFY(socketDescriptor > 0);

    // A socket device is initialized on the server side, passing the

    // socket descriptor from accept(). It's pre-connected.

    QSocketLayer serverSocket;

    QVERIFY(serverSocket.initialize(socketDescriptor));

    QVERIFY(serverSocket.state() == QAbstractSocket::ConnectedState);

    const int messageSize = 1024 * 256;

    QByteArray message1(messageSize, '@');

    QByteArray answer(messageSize, '@');

    QTime timer;

    timer.start();

    qlonglong readBytes = 0;

    while (timer.elapsed() < 5000) {

        qlonglong written = serverSocket.write(message1.data(), message1.size());

        while (written > 0) {

            client.waitForRead();

            if (client.bytesAvailable() > 0) {

                qlonglong readNow = client.read(answer.data(), answer.size());

                written -= readNow;

                readBytes += readNow;

            }

        }

    }

    qDebug("\t\t%.1fMB/%.1fs: %.1fMB/s",

           readBytes / (1024.0 * 1024.0),

           timer.elapsed() / 1024.0,

           (readBytes / (timer.elapsed() / 1000.0)) / (1024 * 1024));

}

*/

void tst_QHttpSocketEngine::tcpSocketBlockingTest()

{

    QHttpSocketEngineHandler http;

    QTcpSocket socket;

    // Connect

    socket.connectToHost(QtNetworkSettings::serverName(), 143);

    QVERIFY(socket.waitForConnected());

    QCOMPARE(socket.state(), QTcpSocket::ConnectedState);

    // Read greeting

    QVERIFY(socket.waitForReadyRead(5000));

    QString s = socket.readLine();

    QCOMPARE(s.toLatin1().constData(), QtNetworkSettings::expectedReplyIMAP().constData());

    // Write NOOP

    QCOMPARE((int) socket.write("1 NOOP\r\n", 8), 8);

    if (!socket.canReadLine())

        QVERIFY(socket.waitForReadyRead(5000));

    // Read response

    s = socket.readLine();

    QCOMPARE(s.toLatin1().constData(), "1 OK Completed\r\n");

    // Write LOGOUT

    QCOMPARE((int) socket.write("2 LOGOUT\r\n", 10), 10);

    if (!socket.canReadLine())

        QVERIFY(socket.waitForReadyRead(5000));

    // Read two lines of respose

    s = socket.readLine();

    QCOMPARE(s.toLatin1().constData(), "* BYE LOGOUT received\r\n");

    if (!socket.canReadLine())

        QVERIFY(socket.waitForReadyRead(5000));

    s = socket.readLine();

    QCOMPARE(s.toLatin1().constData(), "2 OK Completed\r\n");

    // Close the socket

    socket.close();

    // Check that it's closed

    QCOMPARE(socket.state(), QTcpSocket::UnconnectedState);

}

//----------------------------------------------------------------------------------

void tst_QHttpSocketEngine::tcpSocketNonBlockingTest()

{

    QHttpSocketEngineHandler http;

    QTcpSocket socket;

    connect(&socket, SIGNAL(hostFound()), SLOT(tcpSocketNonBlocking_hostFound()));

    connect(&socket, SIGNAL(connected()), SLOT(tcpSocketNonBlocking_connected()));

    connect(&socket, SIGNAL(disconnected()), SLOT(tcpSocketNonBlocking_closed()));

    connect(&socket, SIGNAL(bytesWritten(qint64)), SLOT(tcpSocketNonBlocking_bytesWritten(qint64)));

    connect(&socket, SIGNAL(readyRead()), SLOT(tcpSocketNonBlocking_readyRead()));

    tcpSocketNonBlocking_socket = &socket;

    // Connect

    socket.connectToHost(QtNetworkSettings::serverName(), 143);

    QCOMPARE(socket.state(), QTcpSocket::HostLookupState);

    QTestEventLoop::instance().enterLoop(30);

    if (QTestEventLoop::instance().timeout()) {

        QFAIL("Timed out");

    }

    if (socket.state() == QTcpSocket::ConnectingState) {

        QTestEventLoop::instance().enterLoop(30);

        if (QTestEventLoop::instance().timeout()) {

            QFAIL("Timed out");

        }

    }

    QCOMPARE(socket.state(), QTcpSocket::ConnectedState);

    QTestEventLoop::instance().enterLoop(30);

    if (QTestEventLoop::instance().timeout()) {

        QFAIL("Timed out");

    }