tstRTTcp-1.cpp@ 36993

最後變更在這個檔案從36993是 32810,由 vboxsync 提交於 14 年前
tcp.cpp: Outlined fix rtTcpClose fix for 3.2.
屬性 svn:eol-style 設為 `native` 屬性 svn:keywords 設為 `Author Date Id Revision`
檔案大小: 6.6 KB

行
1	#include <iprt/tcp.h>
2
3	#include <iprt/string.h>
4	#include <iprt/test.h>
5
6
7	/*******************************************************************************
8	* Global Variables *
9	*******************************************************************************/
10	static RTTEST g_hTest;
11
12
13	/* * * * * * * * Test 3 * * * * * * * */
14
15	static DECLCALLBACK(int) test3Server(RTSOCKET hSocket, void *pvUser)
16	{
17	RTTestSetDefault(g_hTest, NULL);
18	char szBuf[4096];
19
20	/* say hello */
21	RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
22	RT_ZERO(szBuf);
23	RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
24	szBuf[sizeof("dude!\n") - 1] = '\0';
25	RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
26
27	/* Send ~20 MB of data that the client receives while trying to disconnect. */
28	RT_ZERO(szBuf);
29	size_t cbSent = 0;
30	while (cbSent < 20 * _1M)
31	{
32	RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, szBuf, sizeof(szBuf)), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
33	cbSent += sizeof(szBuf);
34	}
35
36	return VERR_TCP_SERVER_STOP;
37	}
38
39
40	void test3()
41	{
42	RTTestSub(g_hTest, "Graceful disconnect");
43
44	uint32_t cStartErrors = RTTestErrorCount(g_hTest);
45	for (unsigned i = 0; i < 100 && cStartErrors == RTTestErrorCount(g_hTest); i++)
46	{
47	PRTTCPSERVER pServer;
48	RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2",
49	test3Server, NULL, &pServer), VINF_SUCCESS);
50
51	int rc;
52	RTSOCKET hSocket;
53	RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
54	if (RT_SUCCESS(rc))
55	{
56	char szBuf[512];
57	RT_ZERO(szBuf);
58	do /* break non-loop */
59	{
60	RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
61	RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
62	RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
63	} while (0);
64
65	RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
66	}
67
68	RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
69	}
70	}
71
72
73	/* * * * * * * * Test 2 * * * * * * * */
74
75	static DECLCALLBACK(int) test2Server(RTSOCKET hSocket, void *pvUser)
76	{
77	RTTestSetDefault(g_hTest, NULL);
78	char szBuf[512];
79
80	/* say hello */
81	RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
82	RT_ZERO(szBuf);
83	RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
84	szBuf[sizeof("dude!\n") - 1] = '\0';
85	RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
86
87	/* wait for a goodbye which doesn't arrive. */
88	RT_ZERO(szBuf);
89	RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("byebye\n") - 1, NULL), VERR_NET_SHUTDOWN, VERR_TCP_SERVER_STOP);
90
91	return VERR_TCP_SERVER_STOP;
92	}
93
94
95	void test2()
96	{
97	RTTestSub(g_hTest, "Rude client");
98
99	PRTTCPSERVER pServer;
100	RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-2",
101	test2Server, NULL, &pServer), VINF_SUCCESS);
102
103	int rc;
104	RTSOCKET hSocket;
105	RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
106	if (RT_SUCCESS(rc))
107	{
108	char szBuf[512];
109	RT_ZERO(szBuf);
110	do /* break non-loop */
111	{
112	RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
113	RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
114	RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
115	} while (0);
116
117	RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
118	}
119
120	RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
121	}
122
123
124	/* * * * * * * * Test 1 * * * * * * * */
125
126	static DECLCALLBACK(int) test1Server(RTSOCKET hSocket, void *pvUser)
127	{
128	RTTestSetDefault(g_hTest, NULL);
129
130	char szBuf[512];
131	RTTESTI_CHECK_RET(pvUser == NULL, VERR_TCP_SERVER_STOP);
132
133	/* say hello */
134	RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "hello\n", sizeof("hello\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
135	RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("dude!\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
136	szBuf[sizeof("dude!\n") - 1] = '\0';
137	RTTESTI_CHECK_RET(strcmp(szBuf, "dude!\n") == 0, VERR_TCP_SERVER_STOP);
138
139	/* say goodbye */
140	RTTESTI_CHECK_RC_RET(RTTcpRead(hSocket, szBuf, sizeof("byebye\n") - 1, NULL), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
141	szBuf[sizeof("byebye\n") - 1] = '\0';
142	RTTESTI_CHECK_RET(strcmp(szBuf, "byebye\n") == 0, VERR_TCP_SERVER_STOP);
143	RTTESTI_CHECK_RC_RET(RTTcpWrite(hSocket, "bye\n", sizeof("bye\n") - 1), VINF_SUCCESS, VERR_TCP_SERVER_STOP);
144
145	return VERR_TCP_SERVER_STOP;
146	}
147
148
149	void test1()
150	{
151	RTTestSub(g_hTest, "Simple server-client setup");
152
153	PRTTCPSERVER pServer;
154	RTTESTI_CHECK_RC_RETV(RTTcpServerCreate("localhost", 9999, RTTHREADTYPE_DEFAULT, "server-1",
155	test1Server, NULL, &pServer), VINF_SUCCESS);
156
157	int rc;
158	RTSOCKET hSocket;
159	RTTESTI_CHECK_RC(rc = RTTcpClientConnect("localhost", 9999, &hSocket), VINF_SUCCESS);
160	if (RT_SUCCESS(rc))
161	{
162	do /* break non-loop */
163	{
164	char szBuf[512];
165	RT_ZERO(szBuf);
166	RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("hello\n") - 1, NULL), VINF_SUCCESS);
167	RTTESTI_CHECK_BREAK(strcmp(szBuf, "hello\n") == 0);
168	RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "dude!\n", sizeof("dude!\n") - 1), VINF_SUCCESS);
169
170	RTTESTI_CHECK_RC_BREAK(RTTcpWrite(hSocket, "byebye\n", sizeof("byebye\n") - 1), VINF_SUCCESS);
171	RT_ZERO(szBuf);
172	RTTESTI_CHECK_RC_BREAK(RTTcpRead(hSocket, szBuf, sizeof("bye\n") - 1, NULL), VINF_SUCCESS);
173	RTTESTI_CHECK_BREAK(strcmp(szBuf, "bye\n") == 0);
174	} while (0);
175
176	RTTESTI_CHECK_RC(RTTcpClientClose(hSocket), VINF_SUCCESS);
177	}
178
179	RTTESTI_CHECK_RC(RTTcpServerDestroy(pServer), VINF_SUCCESS);
180	}
181
182
183	int main()
184	{
185	RTEXITCODE rcExit = RTTestInitAndCreate("tstRTTcp-1", &g_hTest);
186	if (rcExit != RTEXITCODE_SUCCESS)
187	return rcExit;
188	RTTestBanner(g_hTest);
189
190	test1();
191	test2();
192	test3();
193
194	/** @todo test the full RTTcp API. */
195
196	return RTTestSummaryAndDestroy(g_hTest);
197	}
198

注意: 瀏覽 TracBrowser 來幫助您使用儲存庫瀏覽器

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

以其他格式下載: