DarkflameServer/thirdparty/raknet/Source/RakNetTransport.cpp

182 lines
4.9 KiB
C++
Raw Normal View History

#include "RakNetTransport.h"
#include "RakNetworkFactory.h"
#include "RakPeerInterface.h"
#include "BitStream.h"
#include "MessageIdentifiers.h"
#include <stdio.h>
#include <string.h>
#include <stdarg.h>
#include "LinuxStrings.h"
#ifdef _MSC_VER
#pragma warning( push )
#endif
RakNetTransportCommandParser::RakNetTransportCommandParser()
{
RegisterCommand(1, "SetPassword","Changes the console password to whatever.");
RegisterCommand(0, "ClearPassword","Removes the console passwords.");
RegisterCommand(0, "GetPassword","Gets the console password.");
}
RakNetTransportCommandParser::~RakNetTransportCommandParser()
{
}
bool RakNetTransportCommandParser::OnCommand(const char *command, unsigned numParameters, char **parameterList, TransportInterface *transport, SystemAddress systemAddress, const char *originalString)
{
(void) originalString;
(void) numParameters;
RakNetTransport *rnt = (RakNetTransport*) transport;
if (strcmp(command, "SetPassword")==0)
{
rnt->SetIncomingPassword(parameterList[0]);
rnt->Send(systemAddress, "Password changed to %s\r\n", parameterList[0]);
}
else if (strcmp(command, "ClearPassword")==0)
{
rnt->SetIncomingPassword(0);
rnt->Send(systemAddress, "Password cleared\r\n");
}
else if (strcmp(command, "GetPassword")==0)
{
char *password;
password=rnt->GetIncomingPassword();
if (password[0])
rnt->Send(systemAddress, "Password is %s\r\n",password);
else
rnt->Send(systemAddress, "No password is set.\r\n");
}
return true;
}
const char *RakNetTransportCommandParser::GetName(void) const
{
return "RakNetTransport";
}
void RakNetTransportCommandParser::SendHelp(TransportInterface *transport, SystemAddress systemAddress)
{
transport->Send(systemAddress, "Provides a secure connection between your console\r\n");
transport->Send(systemAddress, "and the console server. Used to modify the console password.\r\n");
}
RakNetTransport::RakNetTransport()
{
rakPeer=0;
}
RakNetTransport::~RakNetTransport()
{
if (rakPeer)
RakNetworkFactory::DestroyRakPeerInterface(rakPeer);
}
bool RakNetTransport::Start(unsigned short port, bool serverMode)
{
AutoAllocate();
rakPeer->InitializeSecurity(0,0,0,0);
if (serverMode)
{
// Allow up to 8 remote systems to login
rakPeer->SetMaximumIncomingConnections(8);
}
SocketDescriptor socketDescriptor(port,0);
return rakPeer->Startup(8, 250, &socketDescriptor, 1);
}
void RakNetTransport::Stop(void)
{
if (rakPeer==0) return;
rakPeer->Shutdown(1000, 0);
newConnections.Clear();
lostConnections.Clear();
}
void RakNetTransport::Send( SystemAddress systemAddress, const char *data, ... )
{
if (rakPeer==0) return;
if (data==0 || data[0]==0) return;
char text[REMOTE_MAX_TEXT_INPUT];
va_list ap;
va_start(ap, data);
_vsnprintf(text, REMOTE_MAX_TEXT_INPUT, data, ap);
va_end(ap);
text[REMOTE_MAX_TEXT_INPUT-1]=0;
RakNet::BitStream str;
str.Write((MessageID)ID_TRANSPORT_STRING);
str.Write(text, (int) strlen(text));
str.Write((unsigned char) 0); // Null terminate the string
rakPeer->Send(&str, MEDIUM_PRIORITY, RELIABLE_ORDERED, 0, systemAddress, (systemAddress==UNASSIGNED_SYSTEM_ADDRESS)!=0);
}
void RakNetTransport::CloseConnection( SystemAddress systemAddress )
{
rakPeer->CloseConnection(systemAddress, true, 0);
}
Packet* RakNetTransport::Receive( void )
{
if (rakPeer==0) return 0;
Packet *p;
p=rakPeer->Receive();
if (p==0)
return 0;
if (p->data[0]==ID_TRANSPORT_STRING)
{
p->data++; // Go past ID_TRANSPORT_STRING, since the transport protocol is only supposed to send strings.
return p;
}
if (p->data[0]==ID_NEW_INCOMING_CONNECTION)
{
newConnections.Push(p->systemAddress);
}
else if (p->data[0]==ID_DISCONNECTION_NOTIFICATION || p->data[0]==ID_CONNECTION_LOST)
{
lostConnections.Push(p->systemAddress);
}
rakPeer->DeallocatePacket(p);
return 0;
}
SystemAddress RakNetTransport::HasNewConnection(void)
{
if (newConnections.Size())
return newConnections.Pop();
return UNASSIGNED_SYSTEM_ADDRESS;
}
SystemAddress RakNetTransport::HasLostConnection(void)
{
if (lostConnections.Size())
return lostConnections.Pop();
return UNASSIGNED_SYSTEM_ADDRESS;
}
void RakNetTransport::SetIncomingPassword(const char *password)
{
if (password)
rakPeer->SetIncomingPassword(password, (int) strlen(password)+1);
else
rakPeer->SetIncomingPassword(0, 0);
}
char * RakNetTransport::GetIncomingPassword(void)
{
static char password[256];
int passwordLength=255;
rakPeer->GetIncomingPassword((char*)password, &passwordLength);
password[passwordLength]=0;
return (char*) password;
}
void RakNetTransport::DeallocatePacket( Packet *packet )
{
if (rakPeer==0) return;
packet->data--; // Go back to ID_TRANSPORT_STRING, which we passed up in Receive()
rakPeer->DeallocatePacket(packet);
}
void RakNetTransport::AutoAllocate(void)
{
if (rakPeer==0)
rakPeer=RakNetworkFactory::GetRakPeerInterface();
}
CommandParserInterface* RakNetTransport::GetCommandParser(void)
{
return &rakNetTransportCommandParser;
}
#ifdef _MSC_VER
#pragma warning( pop )
#endif