From 48a4a67e8de835d4811560dc99398e365317d490 Mon Sep 17 00:00:00 2001
From: Alex Trotsenko <alex1973tr@gmail.com>
Date: Mon, 4 Aug 2014 15:19:37 +0300
Subject: Fix QAbstractSocket::readData() behavior on buffered socket

Remove an useless check which spontaneously allow direct reads from the
socket engine.

Change-Id: Ia3d2a572d6f1563d613fe2f00d0d6849df259827
Reviewed-by: Oswald Buddenhagen <oswald.buddenhagen@digia.com>
Reviewed-by: Thiago Macieira <thiago.macieira@intel.com>
---
 .../socket/qabstractsocket/tst_qabstractsocket.cpp | 53 ++++++++++++++++++++++
 1 file changed, 53 insertions(+)

(limited to 'tests')

diff --git a/tests/auto/network/socket/qabstractsocket/tst_qabstractsocket.cpp b/tests/auto/network/socket/qabstractsocket/tst_qabstractsocket.cpp
index 28e7c30544..5b314e4e77 100644
--- a/tests/auto/network/socket/qabstractsocket/tst_qabstractsocket.cpp
+++ b/tests/auto/network/socket/qabstractsocket/tst_qabstractsocket.cpp
@@ -45,6 +45,11 @@
 #include <qcoreapplication.h>
 #include <qdebug.h>
 #include <qabstractsocket.h>
+#include <qtcpserver.h>
+#include <qtcpsocket.h>
+#ifndef QT_NO_SSL
+#include <qsslsocket.h>
+#endif
 
 class tst_QAbstractSocket : public QObject
 {
@@ -55,7 +60,9 @@ public:
     virtual ~tst_QAbstractSocket();
 
 private slots:
+    void initTestCase();
     void getSetCheck();
+    void serverDisconnectWithBuffered();
 };
 
 tst_QAbstractSocket::tst_QAbstractSocket()
@@ -74,6 +81,11 @@ public:
     void setPeerPort(quint16 port) { QAbstractSocket::setPeerPort(port); }
 };
 
+void tst_QAbstractSocket::initTestCase()
+{
+    qRegisterMetaType<QAbstractSocket::SocketState>("QAbstractSocket::SocketState");
+}
+
 // Testing get/set functions
 void tst_QAbstractSocket::getSetCheck()
 {
@@ -102,5 +114,46 @@ void tst_QAbstractSocket::getSetCheck()
     QCOMPARE(quint16(0xffff), obj1.peerPort());
 }
 
+// Test buffered socket being properly closed on remote disconnect
+void tst_QAbstractSocket::serverDisconnectWithBuffered()
+{
+    QTcpServer tcpServer;
+#ifndef QT_NO_SSL
+    QSslSocket testSocket;
+#else
+    QTcpSocket testSocket;
+#endif
+
+    QVERIFY(tcpServer.listen(QHostAddress::LocalHost));
+    testSocket.connectToHost(tcpServer.serverAddress(), tcpServer.serverPort());
+    // Accept connection on server side
+    QVERIFY(tcpServer.waitForNewConnection(5000));
+    QTcpSocket *newConnection = tcpServer.nextPendingConnection();
+    // Send one char and drop link
+    QVERIFY(newConnection != NULL);
+    QVERIFY(newConnection->putChar(0));
+    QVERIFY(newConnection->flush());
+    delete newConnection;
+
+    QVERIFY(testSocket.waitForConnected(5000)); // ready for write
+    QVERIFY(testSocket.state() == QAbstractSocket::ConnectedState);
+
+    QSignalSpy spyStateChanged(&testSocket, SIGNAL(stateChanged(QAbstractSocket::SocketState)));
+    QSignalSpy spyDisconnected(&testSocket, SIGNAL(disconnected()));
+
+    QVERIFY(testSocket.waitForReadyRead(5000)); // have one char already in internal buffer
+    char buf[128];
+    QCOMPARE(testSocket.read(buf, sizeof(buf)), Q_INT64_C(1));
+    if (testSocket.state() != QAbstractSocket::UnconnectedState) {
+        QVERIFY(testSocket.waitForDisconnected(5000));
+        QVERIFY(testSocket.state() == QAbstractSocket::UnconnectedState);
+    }
+    // Test signal emitting
+    QVERIFY(spyDisconnected.count() == 1);
+    QVERIFY(spyStateChanged.count() > 0);
+    QVERIFY(qvariant_cast<QAbstractSocket::SocketState>(spyStateChanged.last().first())
+            == QAbstractSocket::UnconnectedState);
+}
+
 QTEST_MAIN(tst_QAbstractSocket)
 #include "tst_qabstractsocket.moc"
-- 
cgit v1.2.3