#include <iprt/tcp.h>

#include <iprt/string.h>

#include <iprt/test.h>

static RTTEST g_hTest;

static DECLCALLBACK(int) test2Server(RTSOCKET hSocket, void *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);

}

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();

/** @todo test the full RTTcp API. */

return RTTestSummaryAndDestroy(g_hTest);

}