bullet3/test/clsocket/RecvAsync.cpp
erwincoumans ab8f16961e Code-style consistency improvement:
Apply clang-format-all.sh using the _clang-format file through all the cpp/.h files.
make sure not to apply it to certain serialization structures, since some parser expects the * as part of the name, instead of type.
This commit contains no other changes aside from adding and applying clang-format-all.sh
2018-09-23 14:17:31 -07:00

101 lines
1.9 KiB
C++

#include <pthread.h>
#include "PassiveSocket.h"
#ifdef WIN32
#include <windows.h>
// usually defined with #include <unistd.h>
static void sleep(unsigned int seconds)
{
Sleep(seconds * 1000);
}
#endif
#define MAX_PACKET 4096
#define TEST_PACKET "Test Packet"
struct thread_data
{
const char *pszServerAddr;
short int nPort;
int nNumBytesToReceive;
int nTotalPayloadSize;
};
void *CreateTCPEchoServer(void *param)
{
CPassiveSocket socket;
CActiveSocket *pClient = NULL;
struct thread_data *pData = (struct thread_data *)param;
int nBytesReceived = 0;
socket.Initialize();
socket.Listen(pData->pszServerAddr, pData->nPort);
if ((pClient = socket.Accept()) != NULL)
{
while (nBytesReceived != pData->nTotalPayloadSize)
{
if (nBytesReceived += pClient->Receive(pData->nNumBytesToReceive))
{
pClient->Send((const uint8 *)pClient->GetData(), pClient->GetBytesReceived());
}
}
sleep(100);
delete pClient;
}
socket.Close();
return NULL;
}
int main(int argc, char **argv)
{
pthread_t threadId;
struct thread_data thData;
CActiveSocket client;
char result[1024];
thData.pszServerAddr = "127.0.0.1";
thData.nPort = 6789;
thData.nNumBytesToReceive = 1;
thData.nTotalPayloadSize = (int)strlen(TEST_PACKET);
pthread_create(&threadId, 0, CreateTCPEchoServer, &thData);
sleep(1); // allow a second for the thread to create and listen
client.Initialize();
client.SetNonblocking();
if (client.Open("127.0.0.1", 6789))
{
if (client.Send((uint8 *)TEST_PACKET, strlen(TEST_PACKET)))
{
int numBytes = -1;
int bytesReceived = 0;
client.Select();
while (bytesReceived != strlen(TEST_PACKET))
{
numBytes = client.Receive(MAX_PACKET);
if (numBytes > 0)
{
bytesReceived += numBytes;
memset(result, 0, 1024);
memcpy(result, client.GetData(), numBytes);
printf("received %d bytes: '%s'\n", numBytes, result);
}
else
{
printf("Received %d bytes\n", numBytes);
}
}
}
}
}