summaryrefslogtreecommitdiffstats
path: root/src/VBox/Runtime/testcase/tstRTTcp-1.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/VBox/Runtime/testcase/tstRTTcp-1.cpp')
-rw-r--r--src/VBox/Runtime/testcase/tstRTTcp-1.cpp246
1 files changed, 246 insertions, 0 deletions
diff --git a/src/VBox/Runtime/testcase/tstRTTcp-1.cpp b/src/VBox/Runtime/testcase/tstRTTcp-1.cpp
new file mode 100644
index 00000000..cb10411f
--- /dev/null
+++ b/src/VBox/Runtime/testcase/tstRTTcp-1.cpp
@@ -0,0 +1,246 @@
+/* $Id: tstRTTcp-1.cpp $ */
+/** @file
+ * IPRT testcase - TCP.
+ */
+
+/*
+ * Copyright (C) 2010-2020 Oracle Corporation
+ *
+ * This file is part of VirtualBox Open Source Edition (OSE), as
+ * available from http://www.virtualbox.org. This file is free software;
+ * you can redistribute it and/or modify it under the terms of the GNU
+ * General Public License (GPL) as published by the Free Software
+ * Foundation, in version 2 as it comes in the "COPYING" file of the
+ * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
+ * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
+ *
+ * The contents of this file may alternatively be used under the terms
+ * of the Common Development and Distribution License Version 1.0
+ * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
+ * VirtualBox OSE distribution, in which case the provisions of the
+ * CDDL are applicable instead of those of the GPL.
+ *
+ * You may elect to license modified versions of this file under the
+ * terms and conditions of either the GPL or the CDDL or both.
+ */
+
+
+/*********************************************************************************************************************************
+* Header Files *
+*********************************************************************************************************************************/
+#include <iprt/tcp.h>
+
+#include <iprt/err.h>
+#include <iprt/string.h>
+#include <iprt/test.h>
+
+
+/*********************************************************************************************************************************
+* Global Variables *
+*********************************************************************************************************************************/
+static RTTEST g_hTest;
+
+
+/* * * * * * * * Test 3 * * * * * * * */
+
+static DECLCALLBACK(int) test3Server(RTSOCKET hSocket, void *pvUser)
+{
+ RT_NOREF_PV(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;
+ int rc = RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2", test3Server, NULL, &pServer);
+#if defined(RT_OS_SOLARIS) || defined(RT_OS_LINUX)
+ /** @todo testboxsh1 occationally hits this for some stupid reason. i=21 in
+ * one occurrence. Fudge a bit for now and see if it helps.
+ * Same for testboxopt, i=98 in another case. */
+ if (rc == VERR_NET_ADDRESS_IN_USE)
+ {
+ RTThreadSleep(500);
+ rc = RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2", test3Server, NULL, &pServer);
+ }
+#endif
+ if (rc != VINF_SUCCESS)
+ {
+ RTTestIFailed("RTTcpServerCreate -> %Rrc, i=%d", rc, i);
+ return;
+ }
+
+ 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)
+{
+ RT_NOREF_PV(pvUser);
+
+ RTTestSetDefault(g_hTest, NULL);
+ char szBuf[512];
+
+ /* 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);
+
+ /* wait for a goodbye which doesn't arrive. */
+ RT_ZERO(szBuf);
+ RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("byebye\n") - 1, NULL), VERR_NET_SHUTDOWN, VERR_TCP_SERVER_STOP);
+
+ return VERR_TCP_SERVER_STOP;
+}
+
+
+void test2()
+{
+ RTTestSub(g_hTest, "Rude client");
+
+ PRTTCPSERVER pServer;
+ RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2",
+ test2Server, 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 1 * * * * * * * */
+
+static DECLCALLBACK(int) test1Server(RTSOCKET hSocket, void *pvUser)
+{
+ RTTestSetDefault(g_hTest, NULL);
+
+ char szBuf[512];
+ RTTESTI_CHECK_RET(pvUser == NULL, VERR_TCP_SERVER_STOP);
+
+ /* say hello */
+ RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
+ 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);
+
+ /* say goodbye */
+ RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("byebye\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
+ szBuf[sizeof("byebye\n") - 1] = '\0';
+ RTTESTI_CHECK_RET(strcmp(szBuf, "byebye\n") == 0, VERR_TCP_SERVER_STOP);
+ RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "bye\n", sizeof("bye\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
+
+ return VERR_TCP_SERVER_STOP;
+}
+
+
+void test1()
+{
+ RTTestSub(g_hTest, "Simple server-client setup");
+
+ PRTTCPSERVER pServer;
+ RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-1",
+ test1Server, NULL, &pServer), VINF_SUCCESS);
+
+ int rc;
+ RTSOCKET hSocket;
+ RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
+ if (RT_SUCCESS(rc))
+ {
+ do /* break non-loop */
+ {
+ char szBuf[512];
+ RT_ZERO(szBuf);
+ 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);
+
+ RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "byebye\n", sizeof("byebye\n") - 1), VINF_SUCCESS);
+ RT_ZERO(szBuf);
+ RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("bye\n") - 1, NULL), VINF_SUCCESS);
+ RTTESTI_CHECK_BREAK(strcmp(szBuf, "bye\n") == 0);
+ } while (0);
+
+ RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
+ }
+
+ RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
+}
+
+
+int main()
+{
+ RTEXITCODE rcExit = RTTestInitAndCreate("tstRTTcp-1", &g_hTest);
+ if (rcExit != RTEXITCODE_SUCCESS)
+ return rcExit;
+ RTTestBanner(g_hTest);
+
+ test1();
+ test2();
+ test3();
+
+ /** @todo test the full RTTcp API. */
+
+ return RTTestSummaryAndDestroy(g_hTest);
+}
+