Changeset 32810 in vbox for trunk/src/VBox/Runtime/testcase

Timestamp:

Sep 29, 2010 10:56:54 AM (14 years ago)

Author:

vboxsync

Message:

tcp.cpp: Outlined fix rtTcpClose fix for 3.2.

File:

• trunk/src/VBox/Runtime/testcase/tstRTTcp-1.cpp (modified) (3 diffs)

trunk/src/VBox/Runtime/testcase/tstRTTcp-1.cpp

@@ -10 +10 @@
 static RTTEST g_hTest;
 
+
+/* * * * * * * *   Test 3    * * * * * * * */
+
+static DECLCALLBACK(int) test3Server(RTSOCKET hSocket, void *pvUser)
+{
+    RTTestSetDefault(g_hTest, NULL);
+    char szBuf[4096];
+
+    /* say hello */
+    RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
+    RT_ZERO(szBuf);
+    RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
+    szBuf[sizeof("dude!\n") - 1] = '\0';
+    RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
+
+    /* Send ~20 MB of data that the client receives while trying to disconnect. */
+    RT_ZERO(szBuf);
+    size_t cbSent = 0;
+    while (cbSent < 20 * _1M)
+    {
+        RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, szBuf, sizeof(szBuf)), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
+        cbSent += sizeof(szBuf);
+    }
+
+    return VERR_TCP_SERVER_STOP;
+}
+
+
+void test3()
+{
+    RTTestSub(g_hTest, "Graceful disconnect");
+
+    uint32_t cStartErrors = RTTestErrorCount(g_hTest);
+    for (unsigned i = 0; i < 100 && cStartErrors == RTTestErrorCount(g_hTest); i++)
+    {
+        PRTTCPSERVER pServer;
+        RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2",
+                                                test3Server, NULL, &pServer), VINF_SUCCESS);
+
+        int rc;
+        RTSOCKET hSocket;
+        RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
+        if (RT_SUCCESS(rc))
+        {
+            char szBuf[512];
+            RT_ZERO(szBuf);
+            do /* break non-loop */
+            {
+                RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
+                RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
+                RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
+            } while (0);
+
+            RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
+        }
+
+        RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
+    }
+}
+
+
+/* * * * * * * *   Test 2    * * * * * * * */
 
 static DECLCALLBACK(int) test2Server(RTSOCKET hSocket, void *pvUser)
@@ -59 +121 @@
 }
 
+
+/* * * * * * * *   Test 1    * * * * * * * */
 
 static DECLCALLBACK(int) test1Server(RTSOCKET hSocket, void *pvUser)
@@ -126 +190 @@
     test1();
     test2();
+    test3();
 
     /** @todo test the full RTTcp API. */