1
0
cuberite-2a/Tools/ProtoProxy/Server.cpp

110 lines
2.1 KiB
C++
Raw Normal View History

// Server.cpp
// Interfaces to the cServer class encapsulating the entire "server"
#include "Globals.h"
#include "Server.h"
#include "Connection.h"
#include "../../src/Logger.h"
cServer::cServer(void)
{
}
2016-08-24 15:45:03 -04:00
int cServer::Init(UInt16 a_ListenPort, UInt16 a_ConnectPort)
{
m_ConnectPort = a_ConnectPort;
#ifdef _WIN32
WSAData wsa;
int res = WSAStartup(0x0202, &wsa);
if (res != 0)
{
LOGERROR("Cannot initialize WinSock: %d", res);
return res;
}
#endif // _WIN32
m_ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (m_ListenSocket < 0)
{
#ifdef _WIN32
int err = WSAGetLastError();
#else
int err = errno;
#endif
LOGERROR("Failed to create listener socket: %d", err);
return err;
}
sockaddr_in local;
memset(&local, 0, sizeof(local));
local.sin_family = AF_INET;
local.sin_addr.s_addr = INADDR_ANY; // All interfaces
local.sin_port = htons(a_ListenPort);
2016-08-24 15:45:03 -04:00
if (bind(m_ListenSocket, reinterpret_cast<const sockaddr *>(&local), sizeof(local)) != 0)
{
#ifdef _WIN32
int err = WSAGetLastError();
#else
int err = errno;
#endif
LOGERROR("Failed to bind listener socket: %d", err);
return err;
}
if (listen(m_ListenSocket, 1) != 0)
{
#ifdef _WIN32
int err = WSAGetLastError();
#else
int err = errno;
#endif
printf("Failed to listen on socket: %d\n", err);
return err;
}
2016-08-24 15:45:03 -04:00
LOGINFO("Listening for client connections on port %d, connecting to server at localhost:%d", a_ListenPort, a_ConnectPort);
LOGINFO("Generating protocol encryption keypair...");
m_PrivateKey.Generate();
m_PublicKeyDER = m_PrivateKey.GetPubKeyDER();
return 0;
}
void cServer::Run(void)
{
2015-02-16 03:27:52 -05:00
LOGINFO("Server running.");
while (true)
{
sockaddr_in Addr;
memset(&Addr, 0, sizeof(Addr));
2014-01-17 07:15:54 -05:00
socklen_t AddrSize = sizeof(Addr);
2016-08-24 15:45:03 -04:00
SOCKET client = accept(m_ListenSocket, reinterpret_cast<sockaddr *>(&Addr), &AddrSize);
if (client == INVALID_SOCKET)
{
2014-01-17 07:15:54 -05:00
printf("accept returned an error: %d; bailing out.\n", SocketError);
return;
}
2015-02-15 14:39:53 -05:00
LOGINFO("Client connected, proxying...");
cConnection Connection(client, *this);
Connection.Run();
2015-02-15 14:39:53 -05:00
LOGINFO("Client disconnected. Ready for another connection.");
}
}