1 files changed, 137 insertions, 0 deletions
diff --git a/source/cBlockingTCPLink.cpp b/source/cBlockingTCPLink.cpp
new file mode 100644
index 000000000..a87809857
--- /dev/null
+++ b/source/cBlockingTCPLink.cpp
@@ -0,0 +1,137 @@
+#include "cBlockingTCPLink.h"
+#include "packets/cPacket.h"
+#include <string>
+
+#include "cMCLogger.h"
+
+#ifndef _WIN32
+#include <cstring>
+#include <errno.h>
+#endif
+
+#ifdef _WIN32
+#define MSG_NOSIGNAL (0)
+#endif
+#ifdef __MACH__
+#define MSG_NOSIGNAL (0)
+#endif
+
+cBlockingTCPLink::cBlockingTCPLink()
+	: m_Socket( 0 )
+{
+}
+
+cBlockingTCPLink::~cBlockingTCPLink()
+{
+	CloseSocket();
+}
+
+void cBlockingTCPLink::CloseSocket()
+{
+	if( m_Socket )
+	{
+		closesocket( m_Socket );
+		m_Socket = 0;
+	}
+}
+
+bool cBlockingTCPLink::Connect( const char* a_Address, unsigned int a_Port )
+{
+	if( m_Socket )
+	{
+		LOGWARN("WARNING: cTCPLink Connect() called while still connected. ALWAYS disconnect before re-connecting!");
+	}
+
+	struct hostent *hp;
+	unsigned int addr;
+	struct sockaddr_in server;
+
+#ifdef _WIN32
+	WSADATA wsaData;
+	int wsaret=WSAStartup(/*0x101*/ MAKEWORD(2, 2),&wsaData);
+
+	if(wsaret!=0)
+	{
+		LOGERROR("cTCPLink: WSAStartup returned error");
+		return false;
+	}
+#endif
+
+	m_Socket=socket(AF_INET,SOCK_STREAM,0);
+#ifdef _WIN32
+	if( m_Socket==INVALID_SOCKET )
+#else
+	if( m_Socket < 0 )
+#endif
+	{
+		LOGERROR("cTCPLink: Invalid socket");
+		m_Socket = 0;
+		return false;
+	}
+
+
+	addr=inet_addr( a_Address );
+	hp=gethostbyaddr((char*)&addr,sizeof(addr),AF_INET);
+	if(hp==NULL)
+	{
+		//LOGWARN("cTCPLink: gethostbyaddr returned NULL");
+		hp = gethostbyname( a_Address );
+		if( hp == NULL )
+		{
+			LOGWARN("cTCPLink: Could not resolve %s", a_Address);
+			CloseSocket();
+			return false;
+		}
+	}
+
+	server.sin_addr.s_addr=*((unsigned long*)hp->h_addr);
+	server.sin_family=AF_INET;
+	server.sin_port=htons( (unsigned short)a_Port );
+	if( connect( m_Socket, (struct sockaddr*)&server, sizeof(server) ) )
+	{
+		LOGWARN("cTCPLink: No response from server (%i)", errno);
+		CloseSocket();
+		return false;
+	}
+
+	return true;
+}
+
+int cBlockingTCPLink::Send( char* a_Data, unsigned int a_Size, int a_Flags /* = 0 */ )
+{
+	if( !m_Socket )
+	{
+		LOGWARN("cBlockingTCPLink: Trying to send data without a valid connection!");
+		return -1;
+	}
+	return cPacket::SendData( m_Socket, a_Data, a_Size, a_Flags );
+}
+
+int cBlockingTCPLink::SendMessage( const char* a_Message, int a_Flags /* = 0 */ )
+{
+	if( !m_Socket )
+	{
+		LOGWARN("cBlockingTCPLink: Trying to send message without a valid connection!");
+		return -1;
+	}
+	return cPacket::SendData( m_Socket, a_Message, strlen(a_Message), a_Flags );
+}
+
+std::string cBlockingTCPLink::ReceiveData()
+{
+	if( !m_Socket ) return "";
+
+	int Received = 0;
+	char Buffer[256];
+	std::string Data;
+	while( (Received = recv(m_Socket, Buffer, 256, 0) ) > 0 )
+	{
+		//LOGINFO("Recv: %i", Received);
+		//LOG("%s", Buffer );
+		Data.append( Buffer, Received );
+		memset( Buffer, 0, 256 );
+	}
+
+	//LOGINFO("Received returned: %i", Received );
+	return Data;
+}