Esempio n. 1
0
ByteArrayPtr StreamSocket::recvBuf()
{
    size_t size = recvSize();
    
    ByteArrayPtr ret = ByteArrayPtr(new ByteArray(size));
    is_.read(&(*ret)[0], size);
    if (is_.bad()) { throw std::string("recvBuf() error"); }
    return ret;
}
Esempio n. 2
0
std::string StreamSocket::recvMsg()
{
    size_t length = recvSize();
    if (length > MAX_CONTROL_MESSAGE_SIZE) {
        std::stringstream ss;
        ss << "length > MAX_CONTROL_MESSAGE_SIZE ";
        ss << length;
        throw ss.str();
    }
    
    char buf2[length];
    is_.read(buf2, length);
    if (is_.bad()) { throw std::string("recvMsg() error"); }
    return std::string(buf2, length);
}
AWE::uint32 TCP_Transfer_ServerThread::runThread()
{
	AWE::Socket::TCP::Protocol* pTCP = AWE::Socket::TCP::create();
	CPPUNIT_ASSERT( pTCP );

	// listen on our port
	AWE::Socket::Address kLocalAddress;
	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			pTCP->resolveLocalAddress(
				&kLocalAddress,
				0/*SocketTestCase::s_iServPort*/,
				(SocketTestCase::s_bTestIP4 ? AF_INET : AF_INET6)
				)
			) == AWE::Socket::Error::Success
		);

	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			pTCP->createSocket( kLocalAddress.getAddressFamily() )
			) == AWE::Socket::Error::Success
		);

	AWE::Socket::StreamProtocolHost< AWE::Socket::TCP::Protocol > kHost( pTCP );
	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			kHost.bind( kLocalAddress )
			) == AWE::Socket::Error::Success
		);
	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			pTCP->getBoundAddress( &kLocalAddress )
			) == AWE::Socket::Error::Success
		);

	AWE::Core::String sAddress;
	AWE::uint16 iPort( 0 );
	kLocalAddress.getAddress( &sAddress );
	kLocalAddress.getPort( &g_iServerPort );
	iPort = g_iServerPort;

	std::cout	<< "[Server] Listening for data on IP \"" << sAddress << "\""
				<< " port " << g_iServerPort
				<< std::endl << std::flush;

	SocketTestCase::s_kSemaphore.release();

	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			kHost.listen()
			) == AWE::Socket::Error::Success
		);

	AWE::Socket::Address kRemoteAddress;
	AWE::Socket::TCP::Protocol* pRemoteConn = 0;
	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			kHost.accept( &pRemoteConn, &kRemoteAddress )
			) == AWE::Socket::Error::Success
		);
	CPPUNIT_ASSERT( pRemoteConn );

	kRemoteAddress.getAddress( &sAddress );
	kRemoteAddress.getPort( &iPort );

	std::cout	<< "[Server] Received connection from IP \"" << sAddress << "\""
				<< " port " << iPort
				<< std::endl << std::flush;

	// receive some data
	const AWE::uint32 dataSize( 100 );
	AWE::int8 pData[ dataSize ];
	memset( pData, 0, dataSize );

	AWE::uint32 recvSize( pRemoteConn->receive( pData, dataSize ) );

	// check we received the correct text
	CPPUNIT_ASSERT( recvSize == g_sText.length() );
	CPPUNIT_ASSERT( strcmp( g_sText.c_str(), pData ) == 0 );

	std::cout	<< "[Server] Received \"" << pData << "\""
				<< std::endl << std::flush;


#if defined( WAIT_BEFORE_DESTROY )
	// release our mutex so the client can run
	// <<-- Unlock
	SocketTestCase::s_kSemaphore.release();
#endif

	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			pRemoteConn->closeSocket()
			) == AWE::Socket::Error::Success
		);
	CPPUNIT_ASSERT(
		AWE::Socket::Error::convertNativeError(
			pTCP->closeSocket()
			) == AWE::Socket::Error::Success
		);

	AWE_SAFE_DELETE( pRemoteConn );
	AWE_SAFE_DELETE( pTCP );

	return 0;
}