129 lines
2.0 KiB
C++
129 lines
2.0 KiB
C++
|
|
#include "Globals.h" // NOTE: MSVC stupidness requires this to be the same across all modules
|
|
|
|
#include "TCPLink.h"
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
#define MSG_NOSIGNAL (0)
|
|
#endif
|
|
#ifdef __MACH__
|
|
#define MSG_NOSIGNAL (0)
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
cTCPLink::cTCPLink()
|
|
: m_Socket( 0 )
|
|
, m_StopEvent( new cEvent() )
|
|
{
|
|
}
|
|
|
|
cTCPLink::~cTCPLink()
|
|
{
|
|
if( m_Socket )
|
|
{
|
|
CloseSocket();
|
|
m_StopEvent->Wait();
|
|
}
|
|
delete m_StopEvent;
|
|
}
|
|
|
|
void cTCPLink::CloseSocket()
|
|
{
|
|
if( m_Socket )
|
|
{
|
|
m_Socket.CloseSocket();
|
|
m_Socket = 0;
|
|
}
|
|
}
|
|
|
|
bool cTCPLink::Connect( const AString & a_Address, unsigned int a_Port )
|
|
{
|
|
if( m_Socket )
|
|
{
|
|
LOGWARN("WARNING: cTCPLink Connect() called while still connected. ALWAYS disconnect before re-connecting!");
|
|
}
|
|
|
|
m_Socket = cSocket::CreateSocket();
|
|
if( !m_Socket.IsValid() )
|
|
{
|
|
LOGERROR("cTCPLink: Failed to create socket");
|
|
return false;
|
|
}
|
|
|
|
if (m_Socket.Connect(a_Address, a_Port) != 0)
|
|
{
|
|
LOGWARN("cTCPLink: Cannot connect to server \"%s\" (%s)", m_Socket.GetLastErrorString().c_str());
|
|
m_Socket.CloseSocket();
|
|
return false;
|
|
}
|
|
|
|
cThread( ReceiveThread, this );
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int cTCPLink::Send(const char * a_Data, unsigned int a_Size, int a_Flags /* = 0 */ )
|
|
{
|
|
(void)a_Flags;
|
|
if (!m_Socket.IsValid())
|
|
{
|
|
LOGWARN("cTCPLink: Trying to send data without a valid connection!");
|
|
return -1;
|
|
}
|
|
return m_Socket.Send(a_Data, a_Size);
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int cTCPLink::SendMessage(const char * a_Message, int a_Flags /* = 0 */ )
|
|
{
|
|
(void)a_Flags;
|
|
if (!m_Socket.IsValid())
|
|
{
|
|
LOGWARN("cTCPLink: Trying to send message without a valid connection!");
|
|
return -1;
|
|
}
|
|
return m_Socket.Send(a_Message, strlen(a_Message));
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cTCPLink::ReceiveThread( void* a_Param)
|
|
{
|
|
cTCPLink* self = (cTCPLink*)a_Param;
|
|
cSocket Socket = self->m_Socket;
|
|
int Received = 0;
|
|
do
|
|
{
|
|
char Data[256];
|
|
Received = Socket.Receive(Data, sizeof(Data), 0);
|
|
self->ReceivedData( Data, ((Received > 0) ? Received : -1) );
|
|
} while ( Received > 0 );
|
|
|
|
LOGINFO("cTCPLink Disconnected (%i)", Received );
|
|
|
|
if (Socket == self->m_Socket)
|
|
{
|
|
self->m_StopEvent->Set();
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|