VirtualBox

source: vbox/trunk/src/VBox/Runtime/testcase/tstRTTcp-1.cpp@ 60257

Last change on this file since 60257 was 57358, checked in by vboxsync, 9 years ago

*: scm cleanup run.
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 8.3 KB
Line 
1/* $Id: tstRTTcp-1.cpp 57358 2015-08-14 15:16:38Z vboxsync $ */
2/** @file
3 * IPRT testcase - TCP.
4 */
5
6/*
7 * Copyright (C) 2010-2015 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 *
17 * The contents of this file may alternatively be used under the terms
18 * of the Common Development and Distribution License Version 1.0
19 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
20 * VirtualBox OSE distribution, in which case the provisions of the
21 * CDDL are applicable instead of those of the GPL.
22 *
23 * You may elect to license modified versions of this file under the
24 * terms and conditions of either the GPL or the CDDL or both.
25 */
26
27
28#include <iprt/tcp.h>
29
30#include <iprt/string.h>
31#include <iprt/test.h>
32
33
34/*********************************************************************************************************************************
35* Global Variables *
36*********************************************************************************************************************************/
37static RTTEST g_hTest;
38
39
40/* * * * * * * * Test 3 * * * * * * * */
41
42static DECLCALLBACK(int) test3Server(RTSOCKET hSocket, void *pvUser)
43{
44 RTTestSetDefault(g_hTest, NULL);
45 char szBuf[4096];
46
47 /* say hello */
48 RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
49 RT_ZERO(szBuf);
50 RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
51 szBuf[sizeof("dude!\n") - 1] = '\0';
52 RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
53
54 /* Send ~20 MB of data that the client receives while trying to disconnect. */
55 RT_ZERO(szBuf);
56 size_t cbSent = 0;
57 while (cbSent < 20 * _1M)
58 {
59 RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, szBuf, sizeof(szBuf)), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
60 cbSent += sizeof(szBuf);
61 }
62
63 return VERR_TCP_SERVER_STOP;
64}
65
66
67void test3()
68{
69 RTTestSub(g_hTest, "Graceful disconnect");
70
71 uint32_t cStartErrors = RTTestErrorCount(g_hTest);
72 for (unsigned i = 0; i < 100 && cStartErrors == RTTestErrorCount(g_hTest); i++)
73 {
74 PRTTCPSERVER pServer;
75 int rc = RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2", test3Server, NULL, &pServer);
76#ifdef RT_OS_SOLARIS
77 /** @todo testboxsh1 occationally hits this for some stupid reason. i=21 in
78 * one occurrence. Fudge a bit for now and see if it helps. */
79 if (rc == VERR_NET_ADDRESS_IN_USE)
80 {
81 RTThreadSleep(500);
82 rc = RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2", test3Server, NULL, &pServer);
83 }
84#endif
85 if (rc != VINF_SUCCESS)
86 {
87 RTTestIFailed("RTTcpServerCreate -> %Rrc, i=%d", rc, i);
88 return;
89 }
90
91 RTSOCKET hSocket;
92 RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
93 if (RT_SUCCESS(rc))
94 {
95 char szBuf[512];
96 RT_ZERO(szBuf);
97 do /* break non-loop */
98 {
99 RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
100 RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
101 RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
102 } while (0);
103
104 RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
105 }
106
107 RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
108 }
109}
110
111
112/* * * * * * * * Test 2 * * * * * * * */
113
114static DECLCALLBACK(int) test2Server(RTSOCKET hSocket, void *pvUser)
115{
116 RTTestSetDefault(g_hTest, NULL);
117 char szBuf[512];
118
119 /* say hello */
120 RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
121 RT_ZERO(szBuf);
122 RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
123 szBuf[sizeof("dude!\n") - 1] = '\0';
124 RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
125
126 /* wait for a goodbye which doesn't arrive. */
127 RT_ZERO(szBuf);
128 RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("byebye\n") - 1, NULL), VERR_NET_SHUTDOWN, VERR_TCP_SERVER_STOP);
129
130 return VERR_TCP_SERVER_STOP;
131}
132
133
134void test2()
135{
136 RTTestSub(g_hTest, "Rude client");
137
138 PRTTCPSERVER pServer;
139 RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2",
140 test2Server, NULL, &pServer), VINF_SUCCESS);
141
142 int rc;
143 RTSOCKET hSocket;
144 RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
145 if (RT_SUCCESS(rc))
146 {
147 char szBuf[512];
148 RT_ZERO(szBuf);
149 do /* break non-loop */
150 {
151 RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
152 RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
153 RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
154 } while (0);
155
156 RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
157 }
158
159 RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
160}
161
162
163/* * * * * * * * Test 1 * * * * * * * */
164
165static DECLCALLBACK(int) test1Server(RTSOCKET hSocket, void *pvUser)
166{
167 RTTestSetDefault(g_hTest, NULL);
168
169 char szBuf[512];
170 RTTESTI_CHECK_RET(pvUser == NULL, VERR_TCP_SERVER_STOP);
171
172 /* say hello */
173 RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
174 RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
175 szBuf[sizeof("dude!\n") - 1] = '\0';
176 RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
177
178 /* say goodbye */
179 RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("byebye\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
180 szBuf[sizeof("byebye\n") - 1] = '\0';
181 RTTESTI_CHECK_RET(strcmp(szBuf, "byebye\n") == 0, VERR_TCP_SERVER_STOP);
182 RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "bye\n", sizeof("bye\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
183
184 return VERR_TCP_SERVER_STOP;
185}
186
187
188void test1()
189{
190 RTTestSub(g_hTest, "Simple server-client setup");
191
192 PRTTCPSERVER pServer;
193 RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-1",
194 test1Server, NULL, &pServer), VINF_SUCCESS);
195
196 int rc;
197 RTSOCKET hSocket;
198 RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
199 if (RT_SUCCESS(rc))
200 {
201 do /* break non-loop */
202 {
203 char szBuf[512];
204 RT_ZERO(szBuf);
205 RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
206 RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
207 RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
208
209 RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "byebye\n", sizeof("byebye\n") - 1), VINF_SUCCESS);
210 RT_ZERO(szBuf);
211 RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("bye\n") - 1, NULL), VINF_SUCCESS);
212 RTTESTI_CHECK_BREAK(strcmp(szBuf, "bye\n") == 0);
213 } while (0);
214
215 RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
216 }
217
218 RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
219}
220
221
222int main()
223{
224 RTEXITCODE rcExit = RTTestInitAndCreate("tstRTTcp-1", &g_hTest);
225 if (rcExit != RTEXITCODE_SUCCESS)
226 return rcExit;
227 RTTestBanner(g_hTest);
228
229 test1();
230 test2();
231 test3();
232
233 /** @todo test the full RTTcp API. */
234
235 return RTTestSummaryAndDestroy(g_hTest);
236}
237
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette