summaryrefslogtreecommitdiffstats
path: root/source/cTCPLink.cpp
blob: 0ade65d552801ec78903ff9e746c15fb88c185ba (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

#include "Globals.h"  // NOTE: MSVC stupidness requires this to be the same across all modules

#include "cTCPLink.h"
#include "cSocket.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();
	}
}