3 files changed, 972 insertions, 0 deletions
diff --git a/tests/auto/qsocks5socketengine/.gitignore b/tests/auto/qsocks5socketengine/.gitignore
new file mode 100644
index 0000000..7d64805
--- /dev/null
+++ b/tests/auto/qsocks5socketengine/.gitignore
@@ -0,0 +1 @@
+tst_qsocks5socketengine
diff --git a/tests/auto/qsocks5socketengine/qsocks5socketengine.pro b/tests/auto/qsocks5socketengine/qsocks5socketengine.pro
new file mode 100644
index 0000000..2949ee2
--- /dev/null
+++ b/tests/auto/qsocks5socketengine/qsocks5socketengine.pro
@@ -0,0 +1,13 @@
+load(qttest_p4)
+SOURCES  += tst_qsocks5socketengine.cpp
+
+
+include(../qnativesocketengine/qsocketengine.pri)
+
+
+MOC_DIR=tmp
+
+QT = core network
+
+
+
diff --git a/tests/auto/qsocks5socketengine/tst_qsocks5socketengine.cpp b/tests/auto/qsocks5socketengine/tst_qsocks5socketengine.cpp
new file mode 100644
index 0000000..86333e0
--- /dev/null
+++ b/tests/auto/qsocks5socketengine/tst_qsocks5socketengine.cpp
@@ -0,0 +1,958 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
+** Contact: Qt Software Information (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 either Technology Preview License Agreement or the
+** Beta Release License Agreement.
+**
+** 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.0, included in the file LGPL_EXCEPTION.txt in this
+** package.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3.0 as published by the Free Software
+** Foundation and appearing in the file LICENSE.GPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU General Public License version 3.0 requirements will be
+** met: http://www.gnu.org/copyleft/gpl.html.
+**
+** If you are unsure which license is appropriate for your use, please
+** contact the sales department at qt-sales@nokia.com.
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+
+#include <QtTest/QtTest>
+
+#include <qcoreapplication.h>
+
+#include <private/qsocks5socketengine_p.h>
+#include <qhostinfo.h>
+#include <qhostaddress.h>
+#include <qtcpsocket.h>
+#include <qhttp.h>
+#include <qauthenticator.h>
+#include <qdebug.h>
+#include <qtcpserver.h>
+#include <qmetatype.h>
+#include <qdebug.h>
+
+#include "../network-settings.h"
+
+Q_DECLARE_METATYPE(QQueue<QByteArray>)
+
+class tst_QSocks5SocketEngine : public QObject, public QAbstractSocketEngineReceiver
+{
+    Q_OBJECT
+
+public:
+    tst_QSocks5SocketEngine();
+    virtual ~tst_QSocks5SocketEngine();
+
+
+public slots:
+    void init();
+    void cleanup();
+private slots:
+    void construction();
+    void errorTest_data();
+    void errorTest();
+    void simpleConnectToIMAP();
+    void simpleErrorsAndStates();
+    void udpTest();
+    void serverTest();
+    void tcpSocketBlockingTest();
+    void tcpSocketNonBlockingTest();
+    void downloadBigFile();
+   // void tcpLoopbackPerformance();
+    void passwordAuth();
+    void passwordAuth2();
+
+protected slots:
+    void tcpSocketNonBlocking_hostFound();
+    void tcpSocketNonBlocking_connected();
+    void tcpSocketNonBlocking_closed();
+    void tcpSocketNonBlocking_readyRead();
+    void tcpSocketNonBlocking_bytesWritten(qint64);
+    void exitLoopSlot();
+    void downloadBigFileSlot();
+    void proxyAuthenticationRequired(const QNetworkProxy &proxy, QAuthenticator *auth);
+
+private:
+    void readNotification() { }
+    void writeNotification() { }
+    void exceptionNotification() { }
+    void connectionNotification() { }
+    QTcpSocket *tcpSocketNonBlocking_socket;
+    QStringList tcpSocketNonBlocking_data;
+    qint64 tcpSocketNonBlocking_totalWritten;
+    QTcpSocket *tmpSocket;
+    qint64 bytesAvailable;
+};
+
+class MiniSocks5Server: public QTcpServer
+{
+    Q_OBJECT
+public:
+    QQueue<QByteArray> responses;
+
+    MiniSocks5Server(const QQueue<QByteArray> r)
+        : responses(r)
+    {
+        listen();
+        connect(this, SIGNAL(newConnection()), SLOT(handleNewConnection()));
+    }
+
+private slots:
+    void handleNewConnection()
+    {
+        QTcpSocket *client = nextPendingConnection();
+        connect(client, SIGNAL(readyRead()), SLOT(handleClientCommand()));
+        client->setProperty("pendingResponses", qVariantFromValue(responses));
+    }
+
+    void handleClientCommand()
+    {
+        // WARNING
+        // this assumes that the client command is received in its entirety
+        // should be ok, since SOCKSv5 commands are rather small
+        QTcpSocket *client = static_cast<QTcpSocket *>(sender());
+        QQueue<QByteArray> pendingResponses =
+            qvariant_cast<QQueue<QByteArray> >(client->property("pendingResponses"));
+        if (pendingResponses.isEmpty())
+            client->disconnectFromHost();
+        else
+            client->write(pendingResponses.dequeue());
+        client->setProperty("pendingResponses", qVariantFromValue(pendingResponses));
+    }
+};
+
+static const char *IMAP_IP = "62.70.27.18";
+
+tst_QSocks5SocketEngine::tst_QSocks5SocketEngine()
+{
+}
+
+tst_QSocks5SocketEngine::~tst_QSocks5SocketEngine()
+{
+
+}
+
+void tst_QSocks5SocketEngine::init()
+{
+    tmpSocket = 0;
+    bytesAvailable = 0;
+}
+
+void tst_QSocks5SocketEngine::cleanup()
+{
+}
+
+//---------------------------------------------------------------------------
+void tst_QSocks5SocketEngine::construction()
+{
+    QSocks5SocketEngine 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_QSocks5SocketEngine::errorTest_data()
+{
+    QTest::addColumn<QString>("hostname");
+    QTest::addColumn<int>("port");
+    QTest::addColumn<QString>("username");
+    QTest::addColumn<QQueue<QByteArray> >("responses");
+    QTest::addColumn<int>("expectedError");
+
+    QQueue<QByteArray> responses;
+    QTest::newRow("proxy-host-not-found") << "this-host-does-not-exist." << 1080 << QString()
+                                          << responses
+                                          << int(QAbstractSocket::ProxyNotFoundError);
+    QTest::newRow("proxy-connection-refused") << "127.0.0.1" << 2 << QString()
+                                              << responses
+                                              << int(QAbstractSocket::ProxyConnectionRefusedError);
+
+#define REPLY(name, contents)                                           \
+    static const char raw_ ## name [] = contents;                       \
+    const QByteArray name = QByteArray::fromRawData(raw_ ## name, sizeof raw_ ## name - 1)
+
+    REPLY(garbage, "\4\4\4\4");
+    // authentication method replies
+    REPLY(noAuthentication, "\5\0");
+    REPLY(passwordAuthentication, "\5\2");
+    REPLY(garbageAuthentication, "\5\177");
+    REPLY(noAcceptableAuthentication, "\5\377");
+    // authentication replies
+    REPLY(authenticationAccepted, "\5\0");
+    REPLY(authenticationNotAccepted, "\5\1");
+    // connection replies
+    REPLY(connectionAccepted, "\5\0\0\4\177\0\0\1\0\100");
+    REPLY(connectionNotAllowed, "\5\2\0");
+    REPLY(networkUnreachable, "\5\3\0");
+    REPLY(hostUnreachable, "\5\4\0");
+    REPLY(connectionRefused, "\5\5\0");
+
+#undef REPLY
+
+    responses << garbage;
+    QTest::newRow("garbage1") << QString() << 0 << QString() << responses
+                              << int(QAbstractSocket::ProxyProtocolError);
+
+    responses.clear();
+    responses << noAuthentication << garbage;
+    QTest::newRow("garbage2") << QString() << 0 << QString() << responses
+                              << int(QAbstractSocket::ProxyProtocolError);
+
+    responses.clear();
+    responses << garbageAuthentication;
+    QTest::newRow("unknown-auth-method") << QString() << 0 << QString()
+                                         << responses
+                                         << int(QAbstractSocket::SocketAccessError);
+
+    responses.clear();
+    responses << noAcceptableAuthentication;
+    QTest::newRow("no-acceptable-authentication") << QString() << 0 << QString()
+                                                  << responses
+                                                  << int(QAbstractSocket::ProxyAuthenticationRequiredError);
+
+    responses.clear();
+    responses << passwordAuthentication << authenticationNotAccepted;
+    QTest::newRow("authentication-required") << QString() << 0 << "foo"
+                                             << responses
+                                             << int(QAbstractSocket::ProxyAuthenticationRequiredError);
+
+    responses.clear();
+    responses << noAuthentication << connectionNotAllowed;
+    QTest::newRow("connection-not-allowed") << QString() << 0 << QString()
+                                            << responses
+                                            << int(QAbstractSocket::SocketAccessError);
+
+    responses.clear();
+    responses << noAuthentication << networkUnreachable;
+    QTest::newRow("network-unreachable") << QString() << 0 << QString()
+                                         << responses
+                                         << int(QAbstractSocket::NetworkError);
+
+    responses.clear();
+    responses << noAuthentication << hostUnreachable;
+    QTest::newRow("host-unreachable") << QString() << 0 << QString()
+                                      << responses
+                                      << int(QAbstractSocket::HostNotFoundError);
+
+    responses.clear();
+    responses << noAuthentication << connectionRefused;
+    QTest::newRow("connection-refused") << QString() << 0 << QString()
+                                        << responses
+                                        << int(QAbstractSocket::ConnectionRefusedError);
+}
+
+void tst_QSocks5SocketEngine::errorTest()
+{
+    QFETCH(QString, hostname);
+    QFETCH(int, port);
+    QFETCH(QString, username);
+    QFETCH(QQueue<QByteArray>, responses);
+    QFETCH(int, expectedError);
+
+    MiniSocks5Server server(responses);
+
+    if (hostname.isEmpty()) {
+        hostname = "127.0.0.1";
+        port = server.serverPort();
+    }
+    QTcpSocket socket;
+    socket.setProxy(QNetworkProxy(QNetworkProxy::Socks5Proxy, 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_QSocks5SocketEngine::simpleConnectToIMAP()
+{
+    QSocks5SocketEngine socketDevice;
+
+    // Initialize device
+    QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));
+    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+
+    socketDevice.setProxy(QNetworkProxy(QNetworkProxy::Socks5Proxy, QtNetworkSettings::serverName(), 1080));
+
+    // Connect to imap.trolltech.com's IP
+    QVERIFY(!socketDevice.connectToHost(QHostAddress(IMAP_IP), 143));
+    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectingState);
+    QVERIFY(socketDevice.waitForWrite());
+    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectedState);
+    QVERIFY(socketDevice.peerAddress() == QHostAddress(IMAP_IP));
+
+    // 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(),
+            "* OK esparsett Cyrus IMAP4 v2.2.8 server ready\r\n");
+
+    // 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;
+    QVERIFY(socketDevice.read(&c, sizeof(c)) == -1);
+    QVERIFY(socketDevice.error() == QAbstractSocket::RemoteHostClosedError);
+    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+}
+
+//---------------------------------------------------------------------------
+void tst_QSocks5SocketEngine::simpleErrorsAndStates()
+{
+    {
+        QSocks5SocketEngine socketDevice;
+
+        // Initialize device
+        QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));
+
+        socketDevice.setProxy(QNetworkProxy(QNetworkProxy::Socks5Proxy, QtNetworkSettings::serverName(), 1080));
+
+        QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+        QVERIFY(!socketDevice.connectToHost(QHostInfo::fromName(QtNetworkSettings::serverName()).addresses().first(), 8088));
+        QVERIFY(socketDevice.state() == QAbstractSocket::ConnectingState);
+        if (socketDevice.waitForWrite(15000)) {
+            QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState ||
+                    socketDevice.state() == QAbstractSocket::ConnectedState);
+        } else {
+            QVERIFY(socketDevice.error() == QAbstractSocket::SocketTimeoutError);
+        }
+    }
+
+}
+
+/*
+//---------------------------------------------------------------------------
+void tst_QSocks5SocketEngine::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
+    QSocks5SocketEngine 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_QSocks5SocketEngine::serverTest()
+{
+    QSocks5SocketEngine server;
+
+    // Initialize a Tcp socket
+    QVERIFY(server.initialize(QAbstractSocket::TcpSocket));
+
+    QNetworkProxy proxy(QNetworkProxy::Socks5Proxy, QtNetworkSettings::serverName(), 1080);
+
+    server.setProxy(proxy);
+
+    // Bind to any port on all interfaces
+    QVERIFY(server.bind(QHostAddress("0.0.0.0"), 0));
+    QVERIFY(server.state() == QAbstractSocket::BoundState);
+
+    // Listen for incoming connections
+    QVERIFY(server.listen());
+    QVERIFY(server.state() == QAbstractSocket::ListeningState);
+
+    // Initialize a Tcp socket
+    QSocks5SocketEngine client;
+    QVERIFY(client.initialize(QAbstractSocket::TcpSocket));
+
+    client.setProxy(proxy);
+
+   // QTest::wait(100000); // ### timing problem on win32
+
+
+    // Connect to our server
+    if (!client.connectToHost(server.localAddress(), server.localPort())) {
+        QVERIFY(client.waitForWrite());
+       // QTest::wait(100); // ### timing problem on win32
+        QVERIFY(client.state() == QAbstractSocket::ConnectedState);
+        //QTest::wait(100);
+    }
+
+    QVERIFY(server.waitForRead());
+
+    // The server accepts the connection
+    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.
+
+    QSocks5SocketEngine serverSocket;
+    QVERIFY(serverSocket.initialize(socketDescriptor));
+    QVERIFY(serverSocket.state() == QAbstractSocket::ConnectedState);
+
+    QVERIFY(serverSocket.localAddress() == client.peerAddress());
+    QVERIFY(serverSocket.localPort() == client.peerPort());
+    // this seems depends on the socks server implementation, especially
+    // when connecting /to/ the socks server /through/ the same socks server
+    //QVERIFY(serverSocket.peerAddress() == client.localAddress());
+    //QVERIFY(serverSocket.peerPort() == client.localPort());
+
+    // The server socket sends a greeting to the client
+    QByteArray greeting = "Greetings!";
+    QVERIFY(serverSocket.write(greeting.data(),
+                              greeting.size()) == greeting.size());
+
+    // The client waits for the greeting to arrive
+    QVERIFY(client.waitForRead());
+    qint64 available = client.bytesAvailable();
+    QVERIFY(available > 0);
+
+    // The client reads the greeting and checks that it's correct
+    QByteArray response;
+    response.resize(available);
+    QVERIFY(client.read(response.data(),
+                       response.size()) == response.size());
+    QCOMPARE(response, greeting);
+}
+
+
+//---------------------------------------------------------------------------
+void tst_QSocks5SocketEngine::udpTest()
+{
+    QSocks5SocketEngine udpSocket;
+
+    // Initialize device #1
+    QVERIFY(udpSocket.initialize(QAbstractSocket::UdpSocket));
+    QVERIFY(udpSocket.isValid());
+
+    QNetworkProxy proxy(QNetworkProxy::Socks5Proxy, QtNetworkSettings::serverName(), 1080);
+
+    udpSocket.setProxy(proxy);
+
+    QVERIFY(udpSocket.protocol() == QAbstractSocket::IPv4Protocol);
+    QVERIFY(udpSocket.socketType() == QAbstractSocket::UdpSocket);
+    QVERIFY(udpSocket.state() == QAbstractSocket::UnconnectedState);
+
+    // Bind #1
+    QVERIFY(udpSocket.bind(QHostAddress("0.0.0.0"), 0));
+    QVERIFY(udpSocket.state() == QAbstractSocket::BoundState);
+    QVERIFY(udpSocket.localPort() != 0);
+
+    // Initialize device #2
+    QSocks5SocketEngine udpSocket2;
+    QVERIFY(udpSocket2.initialize(QAbstractSocket::UdpSocket));
+
+    udpSocket2.setProxy(proxy);
+
+    // Connect device #2 to #1
+    QVERIFY(udpSocket2.connectToHost(udpSocket.localAddress(), udpSocket.localPort()));
+    QVERIFY(udpSocket2.state() == QAbstractSocket::ConnectedState);
+
+    // Write a message to #1
+    QByteArray message1 = "hei der";
+    QVERIFY(udpSocket2.write(message1.data(),
+                            message1.size()) == message1.size());
+
+    // Read the message from #2
+    QVERIFY(udpSocket.waitForRead());
+    QVERIFY(udpSocket.hasPendingDatagrams());
+    qint64 available = udpSocket.pendingDatagramSize();
+    QVERIFY(available > 0);
+    QByteArray answer;
+    answer.resize(available);
+    QHostAddress senderAddress;
+    quint16 senderPort = 0;
+    QVERIFY(udpSocket.readDatagram(answer.data(), answer.size(),
+                                  &senderAddress,
+                                  &senderPort) == message1.size());
+    QVERIFY(senderAddress == udpSocket2.localAddress());
+    QVERIFY(senderPort == udpSocket2.localPort());
+}
+
+void tst_QSocks5SocketEngine::tcpSocketBlockingTest()
+{
+    QSocks5SocketEngineHandler socks5;
+
+    QTcpSocket socket;
+
+    // Connect
+    socket.connectToHost("imap.troll.no", 143);
+    QVERIFY(socket.waitForConnected());
+    QCOMPARE(socket.state(), QTcpSocket::ConnectedState);
+
+    // Read greeting
+    QVERIFY(socket.waitForReadyRead(5000));
+    QString s = socket.readLine();
+    QCOMPARE(s.toLatin1().constData(), "* OK esparsett Cyrus IMAP4 v2.2.8 server ready\r\n");
+
+    // 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_QSocks5SocketEngine::tcpSocketNonBlockingTest()
+{
+    QSocks5SocketEngineHandler socks5;
+
+    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("imap.troll.no", 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");
+    }
+
+    // Read greeting
+    QVERIFY(!tcpSocketNonBlocking_data.isEmpty());
+    QCOMPARE(tcpSocketNonBlocking_data.at(0).toLatin1().constData(),
+            "* OK esparsett Cyrus IMAP4 v2.2.8 server ready\r\n");
+    tcpSocketNonBlocking_data.clear();
+
+    tcpSocketNonBlocking_totalWritten = 0;
+
+    // Write NOOP
+    QCOMPARE((int) socket.write("1 NOOP\r\n", 8), 8);
+
+
+    QTestEventLoop::instance().enterLoop(30);
+    if (QTestEventLoop::instance().timeout()) {
+        QFAIL("Timed out");
+    }
+
+    QVERIFY(tcpSocketNonBlocking_totalWritten == 8);
+
+
+    QTestEventLoop::instance().enterLoop(30);
+    if (QTestEventLoop::instance().timeout()) {
+        QFAIL("Timed out");
+    }
+
+
+    // Read response
+    QVERIFY(!tcpSocketNonBlocking_data.isEmpty());
+    QCOMPARE(tcpSocketNonBlocking_data.at(0).toLatin1().constData(), "1 OK Completed\r\n");
+    tcpSocketNonBlocking_data.clear();
+
+
+    tcpSocketNonBlocking_totalWritten = 0;
+
+    // Write LOGOUT
+    QCOMPARE((int) socket.write("2 LOGOUT\r\n", 10), 10);
+
+    QTestEventLoop::instance().enterLoop(30);
+    if (QTestEventLoop::instance().timeout()) {
+        QFAIL("Timed out");
+    }
+
+    QVERIFY(tcpSocketNonBlocking_totalWritten == 10);
+
+    // Wait for greeting
+    QTestEventLoop::instance().enterLoop(30);
+    if (QTestEventLoop::instance().timeout()) {
+        QFAIL("Timed out");
+    }
+
+    // Read two lines of respose
+    QCOMPARE(tcpSocketNonBlocking_data.at(0).toLatin1().constData(), "* BYE LOGOUT received\r\n");
+    QCOMPARE(tcpSocketNonBlocking_data.at(1).toLatin1().constData(), "2 OK Completed\r\n");
+    tcpSocketNonBlocking_data.clear();
+
+    // Close the socket
+    socket.close();
+
+    // Check that it's closed
+    QCOMPARE(socket.state(), QTcpSocket::UnconnectedState);
+}
+
+void tst_QSocks5SocketEngine::tcpSocketNonBlocking_hostFound()
+{
+    QTestEventLoop::instance().exitLoop();
+}
+
+void tst_QSocks5SocketEngine::tcpSocketNonBlocking_connected()
+{
+    QTestEventLoop::instance().exitLoop();
+}
+
+void tst_QSocks5SocketEngine::tcpSocketNonBlocking_readyRead()
+{
+    while (tcpSocketNonBlocking_socket->canReadLine())
+        tcpSocketNonBlocking_data.append(tcpSocketNonBlocking_socket->readLine());
+
+    QTestEventLoop::instance().exitLoop();
+}
+
+void tst_QSocks5SocketEngine::tcpSocketNonBlocking_bytesWritten(qint64 written)
+{
+    tcpSocketNonBlocking_totalWritten += written;
+    QTestEventLoop::instance().exitLoop();
+}
+
+void tst_QSocks5SocketEngine::tcpSocketNonBlocking_closed()
+{
+}
+
+//----------------------------------------------------------------------------------
+
+void tst_QSocks5SocketEngine::downloadBigFile()
+{
+    QSocks5SocketEngineHandler socks5;
+
+    if (tmpSocket)
+        delete tmpSocket;
+    tmpSocket = new QTcpSocket;
+
+    connect(tmpSocket, SIGNAL(connected()), SLOT(exitLoopSlot()));
+    connect(tmpSocket, SIGNAL(readyRead()), SLOT(downloadBigFileSlot()));
+
+    tmpSocket->connectToHost(QtNetworkSettings::serverName(), 80);
+
+    QTestEventLoop::instance().enterLoop(30);
+    if (QTestEventLoop::instance().timeout())
+        QFAIL("Network operation timed out");
+
+    QByteArray hostName = QtNetworkSettings::serverName().toLatin1();
+    QVERIFY(tmpSocket->state() == QAbstractSocket::ConnectedState);
+    QVERIFY(tmpSocket->write("GET /mediumfile HTTP/1.0\r\n") > 0);
+    QVERIFY(tmpSocket->write("HOST: ") > 0);
+    QVERIFY(tmpSocket->write(hostName.data()) > 0);
+    QVERIFY(tmpSocket->write("\r\n") > 0);
+    QVERIFY(tmpSocket->write("\r\n") > 0);
+
+    bytesAvailable = 0;
+
+    QTime stopWatch;
+    stopWatch.start();
+
+#if !defined(Q_OS_WINCE)
+    QTestEventLoop::instance().enterLoop(60);
+#else
+    QTestEventLoop::instance().enterLoop(180);
+#endif
+    if (QTestEventLoop::instance().timeout())
+        QFAIL("Network operation timed out");
+
+    QCOMPARE(bytesAvailable, qint64(10000309));
+
+    QVERIFY(tmpSocket->state() == QAbstractSocket::ConnectedState);
+
+    qDebug("\t\t%.1fMB/%.1fs: %.1fMB/s",
+           bytesAvailable / (1024.0 * 1024.0),
+           stopWatch.elapsed() / 1024.0,
+           (bytesAvailable / (stopWatch.elapsed() / 1000.0)) / (1024 * 1024));
+
+    delete tmpSocket;
+    tmpSocket = 0;
+}
+
+void tst_QSocks5SocketEngine::exitLoopSlot()
+{
+    QTestEventLoop::instance().exitLoop();
+}
+
+
+void tst_QSocks5SocketEngine::downloadBigFileSlot()
+{
+    bytesAvailable += tmpSocket->readAll().size();
+    if (bytesAvailable >= 10000000)
+        QTestEventLoop::instance().exitLoop();
+}
+
+void tst_QSocks5SocketEngine::passwordAuth()
+{
+    QSocks5SocketEngine socketDevice;
+
+    // Initialize device
+    QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));
+    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+
+    socketDevice.setProxy(QNetworkProxy(QNetworkProxy::Socks5Proxy, QtNetworkSettings::serverName(), 1080, "qsockstest", "password"));
+
+    // Connect to imap.trolltech.com's IP
+    QVERIFY(!socketDevice.connectToHost(QHostAddress(IMAP_IP), 143));
+    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectingState);
+    QVERIFY(socketDevice.waitForWrite());
+    if (!socketDevice.connectToHost(QHostAddress(IMAP_IP), 143)) {
+        qDebug("%d, %s", socketDevice.error(), socketDevice.errorString().toLatin1().constData());
+    }
+    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectedState);
+    QVERIFY(socketDevice.peerAddress() == QHostAddress(IMAP_IP));
+
+    // 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(),
+            "* OK esparsett Cyrus IMAP4 v2.2.8 server ready\r\n");
+
+    // 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;
+    QVERIFY(socketDevice.read(&c, sizeof(c)) == -1);
+    QVERIFY(socketDevice.error() == QAbstractSocket::RemoteHostClosedError);
+    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+}
+
+//----------------------------------------------------------------------------------
+
+void tst_QSocks5SocketEngine::proxyAuthenticationRequired(const QNetworkProxy &,
+                                                          QAuthenticator *auth)
+{
+    auth->setUser("qsockstest");
+    auth->setPassword("password");
+}
+
+void tst_QSocks5SocketEngine::passwordAuth2()
+{
+    QSocks5SocketEngine socketDevice;
+
+    // Initialize device
+    QVERIFY(socketDevice.initialize(QAbstractSocket::TcpSocket, QAbstractSocket::IPv4Protocol));
+    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+
+    socketDevice.setProxy(QNetworkProxy(QNetworkProxy::Socks5Proxy, QtNetworkSettings::serverName(), 1081));
+    socketDevice.setReceiver(this);
+
+    // Connect to imap.trolltech.com's IP
+    QVERIFY(!socketDevice.connectToHost(QHostAddress(IMAP_IP), 143));
+    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectingState);
+    while (socketDevice.state() == QAbstractSocket::ConnectingState) {
+        QVERIFY(socketDevice.waitForWrite());
+        socketDevice.connectToHost(QHostAddress(IMAP_IP), 143);
+    }
+    if (socketDevice.state() != QAbstractSocket::ConnectedState)
+        qDebug("%d, %s", socketDevice.error(), socketDevice.errorString().toLatin1().constData());
+    QVERIFY(socketDevice.state() == QAbstractSocket::ConnectedState);
+    QVERIFY(socketDevice.peerAddress() == QHostAddress(IMAP_IP));
+
+    // 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(),
+            "* OK esparsett Cyrus IMAP4 v2.2.8 server ready\r\n");
+
+    // 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;
+    QVERIFY(socketDevice.read(&c, sizeof(c)) == -1);
+    QVERIFY(socketDevice.error() == QAbstractSocket::RemoteHostClosedError);
+    QVERIFY(socketDevice.state() == QAbstractSocket::UnconnectedState);
+}
+
+//----------------------------------------------------------------------------------
+
+QTEST_MAIN(tst_QSocks5SocketEngine)
+#include "tst_qsocks5socketengine.moc"