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