106 lines
2.1 KiB
C++
106 lines
2.1 KiB
C++
|
|
// 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)
|
|
{
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
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);
|
|
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;
|
|
}
|
|
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)
|
|
{
|
|
LOGINFO("Server running.");
|
|
while (true)
|
|
{
|
|
sockaddr_in Addr;
|
|
memset(&Addr, 0, sizeof(Addr));
|
|
socklen_t AddrSize = sizeof(Addr);
|
|
SOCKET client = accept(m_ListenSocket, reinterpret_cast<sockaddr *>(&Addr), &AddrSize);
|
|
if (client == INVALID_SOCKET)
|
|
{
|
|
printf("accept returned an error: %d; bailing out.\n", SocketError);
|
|
return;
|
|
}
|
|
LOGINFO("Client connected, proxying...");
|
|
cConnection Connection(client, *this);
|
|
Connection.Run();
|
|
LOGINFO("Client disconnected. Ready for another connection.");
|
|
}
|
|
}
|