1406 lines
42 KiB
C++
1406 lines
42 KiB
C++
// Copyright (C) 2022 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
|
|
// This file is part of the "Nazara Engine - Network module"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
/*
|
|
Copyright(c) 2002 - 2016 Lee Salzman
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files(the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and / or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions :
|
|
|
|
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
#include <Nazara/Network/ENetHost.hpp>
|
|
#include <Nazara/Math/Algorithm.hpp>
|
|
#include <Nazara/Network/Algorithm.hpp>
|
|
#include <Nazara/Network/ENetPeer.hpp>
|
|
#include <Nazara/Network/NetPacket.hpp>
|
|
#include <Nazara/Utils/OffsetOf.hpp>
|
|
#include <Nazara/Network/Debug.hpp>
|
|
|
|
namespace Nz
|
|
{
|
|
namespace
|
|
{
|
|
static std::size_t s_enetCommandSizes[ENetProtocolCommand_Count] =
|
|
{
|
|
0,
|
|
sizeof(ENetProtocolAcknowledge),
|
|
sizeof(ENetProtocolConnect),
|
|
sizeof(ENetProtocolVerifyConnect),
|
|
sizeof(ENetProtocolDisconnect),
|
|
sizeof(ENetProtocolPing),
|
|
sizeof(ENetProtocolSendReliable),
|
|
sizeof(ENetProtocolSendUnreliable),
|
|
sizeof(ENetProtocolSendFragment),
|
|
sizeof(ENetProtocolSendUnsequenced),
|
|
sizeof(ENetProtocolBandwidthLimit),
|
|
sizeof(ENetProtocolThrottleConfigure),
|
|
sizeof(ENetProtocolSendFragment)
|
|
};
|
|
}
|
|
|
|
|
|
ENetPacketRef ENetHost::AllocatePacket(ENetPacketFlags flags)
|
|
{
|
|
std::size_t poolIndex;
|
|
|
|
ENetPacket* packet = m_packetPool.Allocate(poolIndex);
|
|
|
|
ENetPacketRef enetPacket(&m_packetPool, packet);
|
|
enetPacket->flags = flags;
|
|
enetPacket->poolIndex = poolIndex;
|
|
enetPacket.m_pool = &m_packetPool;
|
|
|
|
return enetPacket;
|
|
}
|
|
|
|
void ENetHost::Broadcast(UInt8 channelId, ENetPacketFlags flags, NetPacket&& packet)
|
|
{
|
|
ENetPacketRef enetPacket = AllocatePacket(flags, std::move(packet));
|
|
|
|
for (ENetPeer& peer : m_peers)
|
|
{
|
|
if (peer.GetState() != ENetPeerState::Connected)
|
|
continue;
|
|
|
|
peer.Send(channelId, enetPacket);
|
|
}
|
|
}
|
|
|
|
bool ENetHost::CheckEvents(ENetEvent* event)
|
|
{
|
|
if (!event)
|
|
return false;
|
|
|
|
event->type = ENetEventType::None;
|
|
event->peer = nullptr;
|
|
event->packet.Reset();
|
|
|
|
return DispatchIncomingCommands(event);
|
|
}
|
|
|
|
ENetPeer* ENetHost::Connect(const IpAddress& remoteAddress, std::size_t channelCount, UInt32 data)
|
|
{
|
|
NazaraAssert(remoteAddress.IsValid(), "Invalid remote address");
|
|
NazaraAssert(remoteAddress.GetPort() != 0, "Remote address has no port");
|
|
|
|
std::size_t peerId;
|
|
for (peerId = 0; peerId < m_peers.size(); ++peerId)
|
|
{
|
|
if (m_peers[peerId].GetState() == ENetPeerState::Disconnected)
|
|
break;
|
|
}
|
|
|
|
if (peerId >= m_peers.size())
|
|
{
|
|
NazaraError("Insufficient peers");
|
|
return nullptr;
|
|
}
|
|
|
|
channelCount = Clamp<std::size_t>(channelCount, ENetConstants::ENetProtocol_MinimumChannelCount, ENetConstants::ENetProtocol_MaximumChannelCount);
|
|
|
|
UInt32 windowSize;
|
|
if (m_outgoingBandwidth == 0)
|
|
windowSize = ENetConstants::ENetProtocol_MaximumWindowSize;
|
|
else
|
|
windowSize = (m_outgoingBandwidth / ENetConstants::ENetPeer_WindowSizeScale) * ENetConstants::ENetProtocol_MinimumWindowSize;
|
|
|
|
ENetPeer& peer = m_peers[peerId];
|
|
peer.InitOutgoing(channelCount, remoteAddress, ++m_randomSeed, windowSize);
|
|
|
|
ENetProtocol command(ENetProtocolCommand_Connect | ENetProtocolFlag_Acknowledge, 0xFF);
|
|
command.connect.channelCount = HostToNet(static_cast<UInt32>(channelCount));
|
|
command.connect.connectID = peer.m_connectID;
|
|
command.connect.data = HostToNet(data);
|
|
command.connect.incomingBandwidth = HostToNet(m_incomingBandwidth);
|
|
command.connect.incomingSessionID = peer.m_incomingSessionID;
|
|
command.connect.mtu = HostToNet(peer.m_mtu);
|
|
command.connect.outgoingBandwidth = HostToNet(m_outgoingBandwidth);
|
|
command.connect.outgoingPeerID = HostToNet(peer.m_incomingPeerID);
|
|
command.connect.outgoingSessionID = peer.m_outgoingSessionID;
|
|
command.connect.packetThrottleAcceleration = HostToNet(peer.m_packetThrottleAcceleration);
|
|
command.connect.packetThrottleDeceleration = HostToNet(peer.m_packetThrottleDeceleration);
|
|
command.connect.packetThrottleInterval = HostToNet(peer.m_packetThrottleInterval);
|
|
command.connect.windowSize = HostToNet(peer.m_windowSize);
|
|
peer.QueueOutgoingCommand(command);
|
|
|
|
return &peer;
|
|
}
|
|
|
|
ENetPeer* ENetHost::Connect(const std::string& hostName, NetProtocol protocol, const std::string& service, ResolveError* error, std::size_t channelCount, UInt32 data)
|
|
{
|
|
std::vector<HostnameInfo> results = IpAddress::ResolveHostname(protocol, hostName, service, error);
|
|
if (results.empty())
|
|
return nullptr;
|
|
|
|
IpAddress hostnameAddress;
|
|
for (const HostnameInfo& result : results)
|
|
{
|
|
if (!result.address)
|
|
continue;
|
|
|
|
hostnameAddress = result.address;
|
|
break; //< Take first valid address
|
|
}
|
|
|
|
if (!hostnameAddress.IsValid())
|
|
{
|
|
if (error)
|
|
*error = ResolveError::NotFound;
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
return Connect(hostnameAddress, channelCount, data);
|
|
}
|
|
|
|
bool ENetHost::Create(const IpAddress& listenAddress, std::size_t peerCount, std::size_t channelCount)
|
|
{
|
|
return Create(listenAddress, peerCount, channelCount, 0, 0);
|
|
}
|
|
|
|
bool ENetHost::Create(const IpAddress& listenAddress, std::size_t peerCount, std::size_t channelCount, UInt32 incomingBandwidth, UInt32 outgoingBandwidth)
|
|
{
|
|
NazaraAssert(listenAddress.IsValid(), "Invalid listening address");
|
|
|
|
if (peerCount > ENetConstants::ENetProtocol_MaximumPeerId)
|
|
{
|
|
NazaraError("Peer count exceeds maximum peer count supported by protocol (" + NumberToString(ENetConstants::ENetProtocol_MaximumPeerId) + ")");
|
|
return false;
|
|
}
|
|
|
|
if (!InitSocket(listenAddress))
|
|
return false;
|
|
|
|
m_address = listenAddress;
|
|
m_allowsIncomingConnections = (listenAddress.IsValid() && !listenAddress.IsLoopback());
|
|
m_randomSeed = *reinterpret_cast<UInt32*>(this);
|
|
m_randomSeed += s_randomGenerator();
|
|
m_randomSeed = (m_randomSeed << 16) | (m_randomSeed >> 16);
|
|
m_channelLimit = Clamp<std::size_t>(channelCount, ENetConstants::ENetProtocol_MinimumChannelCount, ENetConstants::ENetProtocol_MaximumChannelCount);
|
|
m_incomingBandwidth = incomingBandwidth;
|
|
m_outgoingBandwidth = outgoingBandwidth;
|
|
m_bandwidthThrottleEpoch = 0;
|
|
m_recalculateBandwidthLimits = false;
|
|
m_mtu = ENetConstants::ENetHost_DefaultMTU;
|
|
m_commandCount = 0;
|
|
m_bufferCount = 0;
|
|
m_peerCount = peerCount;
|
|
m_receivedAddress = IpAddress::AnyIpV4;
|
|
m_receivedData = nullptr;
|
|
m_receivedDataLength = 0;
|
|
|
|
m_totalSentData = 0;
|
|
m_totalSentPackets = 0;
|
|
m_totalReceivedData = 0;
|
|
m_totalReceivedPackets = 0;
|
|
|
|
m_bandwidthLimitedPeers = 0;
|
|
m_connectedPeers = 0;
|
|
m_duplicatePeers = ENetConstants::ENetProtocol_MaximumPeerId;
|
|
m_maximumPacketSize = ENetConstants::ENetHost_DefaultMaximumPacketSize;
|
|
m_maximumWaitingData = ENetConstants::ENetHost_DefaultMaximumWaitingData;
|
|
|
|
m_peers.reserve(peerCount);
|
|
for (std::size_t i = 0; i < peerCount; ++i)
|
|
m_peers.emplace_back(this, UInt16(i));
|
|
|
|
return true;
|
|
}
|
|
|
|
void ENetHost::Flush()
|
|
{
|
|
UpdateServiceTime();
|
|
|
|
SendOutgoingCommands(nullptr, false);
|
|
}
|
|
|
|
int ENetHost::Service(ENetEvent* event, UInt32 timeout)
|
|
{
|
|
if (event)
|
|
{
|
|
event->type = ENetEventType::None;
|
|
event->peer = nullptr;
|
|
event->packet.Reset();
|
|
|
|
if (DispatchIncomingCommands(event))
|
|
return 1;
|
|
}
|
|
|
|
UpdateServiceTime();
|
|
|
|
timeout += m_serviceTime;
|
|
|
|
do
|
|
{
|
|
if (ENetTimeDifference(m_serviceTime, m_bandwidthThrottleEpoch) >= ENetConstants::ENetHost_BandwidthThrottleInterval)
|
|
ThrottleBandwidth();
|
|
|
|
switch (SendOutgoingCommands(event, true))
|
|
{
|
|
case 1:
|
|
return 1;
|
|
|
|
case -1:
|
|
NazaraError("Error sending outgoing packets");
|
|
return -1;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
// Receiving on an unbound socket which has never sent data is an invalid operation
|
|
if (!m_allowsIncomingConnections && m_totalSentData == 0)
|
|
return 0;
|
|
|
|
switch (ReceiveIncomingCommands(event))
|
|
{
|
|
case 1:
|
|
return 1;
|
|
|
|
case -1:
|
|
//NazaraError("Error receiving incoming packets");
|
|
return -1;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
switch (SendOutgoingCommands(event, true))
|
|
{
|
|
case 1:
|
|
return 1;
|
|
|
|
case -1:
|
|
NazaraError("Error sending outgoing packets");
|
|
|
|
return -1;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (event)
|
|
{
|
|
if (DispatchIncomingCommands(event))
|
|
return 1;
|
|
}
|
|
|
|
if (ENetTimeGreaterEqual(m_serviceTime, timeout))
|
|
return 0;
|
|
|
|
for (;;)
|
|
{
|
|
UpdateServiceTime();
|
|
|
|
if (ENetTimeGreaterEqual(m_serviceTime, timeout))
|
|
return 0;
|
|
|
|
if (m_poller.Wait(ENetTimeDifference(timeout, m_serviceTime)))
|
|
break;
|
|
}
|
|
|
|
UpdateServiceTime();
|
|
}
|
|
while (m_poller.IsReadyToRead(m_socket));
|
|
|
|
return 0;
|
|
}
|
|
|
|
void ENetHost::SimulateNetwork(double packetLossProbability, UInt16 minDelay, UInt16 maxDelay)
|
|
{
|
|
NazaraAssert(maxDelay >= minDelay, "Maximum delay cannot be greater than minimum delay");
|
|
|
|
if (packetLossProbability <= 0.0 && minDelay == 0 && maxDelay == 0)
|
|
m_isSimulationEnabled = false;
|
|
else
|
|
{
|
|
m_isSimulationEnabled = true;
|
|
m_packetDelayDistribution = std::uniform_int_distribution<UInt16>(minDelay, maxDelay);
|
|
m_packetLossProbability = std::bernoulli_distribution(packetLossProbability);
|
|
}
|
|
}
|
|
|
|
bool ENetHost::InitSocket(const IpAddress& address)
|
|
{
|
|
if (!m_socket.Create((m_isUsingDualStack) ? NetProtocol::Any : address.GetProtocol()))
|
|
return false;
|
|
|
|
m_socket.EnableBlocking(false);
|
|
m_socket.EnableBroadcasting(true);
|
|
m_socket.SetReceiveBufferSize(ENetConstants::ENetHost_ReceiveBufferSize);
|
|
m_socket.SetSendBufferSize(ENetConstants::ENetHost_SendBufferSize);
|
|
|
|
if (address.IsValid() && !address.IsLoopback())
|
|
{
|
|
if (m_socket.Bind(address) != SocketState::Bound)
|
|
{
|
|
NazaraError("Failed to bind address " + address.ToString());
|
|
return false;
|
|
}
|
|
}
|
|
|
|
m_poller.RegisterSocket(m_socket, SocketPollEvent::Read);
|
|
|
|
return true;
|
|
}
|
|
|
|
void ENetHost::AddToDispatchQueue(ENetPeer* peer)
|
|
{
|
|
m_dispatchQueue.UnboundedSet(peer->GetPeerId());
|
|
}
|
|
|
|
void ENetHost::RemoveFromDispatchQueue(ENetPeer* peer)
|
|
{
|
|
m_dispatchQueue.UnboundedReset(peer->GetPeerId());
|
|
}
|
|
|
|
bool ENetHost::DispatchIncomingCommands(ENetEvent* event)
|
|
{
|
|
for (std::size_t bit = m_dispatchQueue.FindFirst(); bit != m_dispatchQueue.npos; bit = m_dispatchQueue.FindNext(bit))
|
|
{
|
|
m_dispatchQueue.Reset(bit);
|
|
|
|
ENetPeer& peer = m_peers[bit];
|
|
switch (peer.GetState())
|
|
{
|
|
case ENetPeerState::ConnectionPending:
|
|
peer.ChangeState(ENetPeerState::Connected);
|
|
|
|
event->type = ENetEventType::IncomingConnect;
|
|
event->peer = &peer;
|
|
event->data = peer.m_eventData;
|
|
return true;
|
|
|
|
case ENetPeerState::ConnectionSucceeded:
|
|
peer.ChangeState(ENetPeerState::Connected);
|
|
|
|
event->type = ENetEventType::OutgoingConnect;
|
|
event->peer = &peer;
|
|
event->data = peer.m_eventData;
|
|
return true;
|
|
|
|
case ENetPeerState::Zombie:
|
|
m_recalculateBandwidthLimits = true;
|
|
|
|
event->type = (peer.m_timedOut) ? ENetEventType::DisconnectTimeout : ENetEventType::Disconnect;
|
|
event->peer = &peer;
|
|
event->data = peer.m_eventData;
|
|
|
|
peer.Reset();
|
|
return true;
|
|
|
|
case ENetPeerState::Connected:
|
|
if (peer.m_dispatchedCommands.empty())
|
|
continue;
|
|
|
|
if (!peer.Receive(&event->packet, &event->channelId))
|
|
continue;
|
|
|
|
event->type = ENetEventType::Receive;
|
|
event->peer = &peer;
|
|
|
|
if (!peer.m_dispatchedCommands.empty())
|
|
AddToDispatchQueue(&peer);
|
|
|
|
return true;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
ENetPeer* ENetHost::HandleConnect(ENetProtocolHeader* /*header*/, ENetProtocol* command)
|
|
{
|
|
if (!m_allowsIncomingConnections)
|
|
return nullptr;
|
|
|
|
UInt32 channelCount = NetToHost(command->connect.channelCount);
|
|
|
|
if (channelCount < ENetConstants::ENetProtocol_MinimumChannelCount || channelCount > ENetConstants::ENetProtocol_MaximumChannelCount)
|
|
return nullptr;
|
|
|
|
std::size_t duplicatePeers = 0;
|
|
ENetPeer* peer = nullptr;
|
|
for (ENetPeer& currentPeer : m_peers)
|
|
{
|
|
if (currentPeer.GetState() == ENetPeerState::Disconnected)
|
|
{
|
|
if (!peer)
|
|
peer = ¤tPeer;
|
|
}
|
|
else if (currentPeer.GetState() != ENetPeerState::Connecting)
|
|
{
|
|
// Compare users without comparing their port
|
|
IpAddress first(currentPeer.m_address);
|
|
first.SetPort(0);
|
|
|
|
IpAddress second(m_receivedAddress);
|
|
second.SetPort(0);
|
|
|
|
if (first == second)
|
|
{
|
|
if (currentPeer.m_address.GetPort() == m_receivedAddress.GetPort() && currentPeer.m_connectID == command->connect.connectID)
|
|
return nullptr;
|
|
|
|
++duplicatePeers;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!peer || duplicatePeers >= m_duplicatePeers)
|
|
return nullptr;
|
|
|
|
channelCount = std::min(channelCount, UInt32(m_channelLimit));
|
|
|
|
peer->InitIncoming(channelCount, m_receivedAddress, command->connect);
|
|
|
|
UInt32 windowSize;
|
|
if (m_incomingBandwidth == 0)
|
|
windowSize = ENetConstants::ENetProtocol_MaximumWindowSize;
|
|
else
|
|
windowSize = (m_incomingBandwidth / ENetConstants::ENetPeer_WindowSizeScale) * ENetConstants::ENetProtocol_MinimumWindowSize;
|
|
|
|
windowSize = std::max(windowSize, NetToHost(command->connect.windowSize));
|
|
windowSize = Clamp<UInt32>(windowSize, ENetConstants::ENetProtocol_MinimumWindowSize, ENetConstants::ENetProtocol_MaximumWindowSize);
|
|
|
|
ENetProtocol verifyCommand(ENetProtocolCommand_VerifyConnect | ENetProtocolFlag_Acknowledge, 0xFF);
|
|
verifyCommand.verifyConnect.connectID = peer->m_connectID;
|
|
verifyCommand.verifyConnect.incomingSessionID = peer->m_outgoingSessionID;
|
|
verifyCommand.verifyConnect.outgoingSessionID = peer->m_incomingSessionID;
|
|
|
|
verifyCommand.verifyConnect.channelCount = HostToNet(channelCount);
|
|
verifyCommand.verifyConnect.incomingBandwidth = HostToNet(m_incomingBandwidth);
|
|
verifyCommand.verifyConnect.mtu = HostToNet(peer->GetMtu());
|
|
verifyCommand.verifyConnect.outgoingBandwidth = HostToNet(m_outgoingBandwidth);
|
|
verifyCommand.verifyConnect.outgoingPeerID = HostToNet(peer->GetPeerId());
|
|
verifyCommand.verifyConnect.packetThrottleAcceleration = HostToNet(peer->GetPacketThrottleAcceleration());
|
|
verifyCommand.verifyConnect.packetThrottleDeceleration = HostToNet(peer->GetPacketThrottleDeceleration());
|
|
verifyCommand.verifyConnect.packetThrottleInterval = HostToNet(peer->GetPacketThrottleInterval());
|
|
verifyCommand.verifyConnect.windowSize = HostToNet(windowSize);
|
|
|
|
peer->QueueOutgoingCommand(verifyCommand);
|
|
|
|
return peer;
|
|
}
|
|
|
|
bool ENetHost::HandleIncomingCommands(ENetEvent* event)
|
|
{
|
|
if (m_receivedDataLength < NazaraOffsetOf(ENetProtocolHeader, sentTime))
|
|
return false;
|
|
|
|
ENetProtocolHeader* header = reinterpret_cast<ENetProtocolHeader*>(m_receivedData.Get());
|
|
|
|
UInt16 peerID = NetToHost(header->peerID);
|
|
UInt8 sessionID = (peerID & ENetProtocolHeaderSessionMask) >> ENetProtocolHeaderSessionShift;
|
|
UInt16 flags = peerID & ENetProtocolHeaderFlag_Mask;
|
|
peerID &= ~(ENetProtocolHeaderFlag_Mask | ENetProtocolHeaderSessionMask);
|
|
|
|
std::size_t headerSize = (flags & ENetProtocolHeaderFlag_SentTime) ? sizeof(ENetProtocolHeader) : NazaraOffsetOf(ENetProtocolHeader, sentTime);
|
|
|
|
ENetPeer* peer;
|
|
if (peerID == ENetConstants::ENetProtocol_MaximumPeerId)
|
|
peer = nullptr;
|
|
else
|
|
{
|
|
if (peerID >= m_peers.size())
|
|
return false;
|
|
else
|
|
{
|
|
peer = &m_peers[peerID];
|
|
|
|
if (peer->GetState() == ENetPeerState::Disconnected || peer->GetState() == ENetPeerState::Zombie)
|
|
return false;
|
|
|
|
if (m_receivedAddress != peer->GetAddress() && peer->GetAddress() != IpAddress::BroadcastIpV4)
|
|
return false;
|
|
|
|
if (peer->m_outgoingPeerID < ENetConstants::ENetProtocol_MaximumPeerId && sessionID != peer->m_incomingSessionID)
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Compression handling
|
|
if (flags & ENetProtocolHeaderFlag_Compressed)
|
|
{
|
|
if (!m_compressor)
|
|
return false;
|
|
|
|
std::size_t newSize = m_compressor->Decompress(peer, m_receivedData + headerSize, m_receivedDataLength - headerSize, m_packetData[1].data() + headerSize, m_packetData[1].size() - headerSize);
|
|
if (newSize == 0 || newSize > m_packetData[1].size() - headerSize)
|
|
return false;
|
|
|
|
std::memcpy(m_packetData[1].data(), header, headerSize);
|
|
m_receivedData = m_packetData[1].data();
|
|
m_receivedDataLength = headerSize + newSize;
|
|
}
|
|
|
|
// Checksum
|
|
|
|
if (peer)
|
|
{
|
|
peer->m_address = m_receivedAddress;
|
|
peer->m_incomingDataTotal += UInt32(m_receivedDataLength);
|
|
peer->m_totalByteReceived += UInt32(m_receivedDataLength);
|
|
}
|
|
|
|
auto commandError = [&]() -> bool
|
|
{
|
|
if (event && event->type != ENetEventType::None)
|
|
return true;
|
|
|
|
return false;
|
|
};
|
|
|
|
UInt8* currentData = m_receivedData + headerSize;
|
|
|
|
while (currentData < &m_receivedData[m_receivedDataLength])
|
|
{
|
|
ENetProtocol* command = reinterpret_cast<ENetProtocol*>(currentData);
|
|
|
|
if (currentData + sizeof(ENetProtocolCommandHeader) > &m_receivedData[m_receivedDataLength])
|
|
break;
|
|
|
|
UInt8 commandNumber = command->header.command & ENetProtocolCommand_Mask;
|
|
if (commandNumber >= ENetProtocolCommand_Count)
|
|
break;
|
|
|
|
std::size_t commandSize = s_enetCommandSizes[commandNumber];
|
|
if (commandSize == 0 || currentData + commandSize > &m_receivedData[m_receivedDataLength])
|
|
break;
|
|
|
|
currentData += commandSize;
|
|
|
|
if (!peer && commandNumber != ENetProtocolCommand_Connect)
|
|
break;
|
|
|
|
command->header.reliableSequenceNumber = NetToHost(command->header.reliableSequenceNumber);
|
|
|
|
switch (commandNumber)
|
|
{
|
|
case ENetProtocolCommand_Acknowledge:
|
|
if (!peer->HandleAcknowledge(command, event))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_Connect:
|
|
if (peer)
|
|
return commandError();
|
|
|
|
peer = HandleConnect(header, command);
|
|
if (!peer)
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_VerifyConnect:
|
|
if (!peer->HandleVerifyConnect(command, event))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_Disconnect:
|
|
if (!peer->HandleDisconnect(command))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_Ping:
|
|
if (!peer->HandlePing(command))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_SendReliable:
|
|
if (!peer->HandleSendReliable(command, ¤tData))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_SendUnreliable:
|
|
if (!peer->HandleSendUnreliable(command, ¤tData))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_SendUnsequenced:
|
|
if (!peer->HandleSendUnsequenced(command, ¤tData))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_SendFragment:
|
|
if (!peer->HandleSendFragment(command, ¤tData))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_BandwidthLimit:
|
|
if (!peer->HandleBandwidthLimit(command))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_ThrottleConfigure:
|
|
if (!peer->HandleThrottleConfigure(command))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
case ENetProtocolCommand_SendUnreliableFragment:
|
|
if (!peer->HandleSendUnreliableFragment(command, ¤tData))
|
|
return commandError();
|
|
|
|
break;
|
|
|
|
default:
|
|
return commandError();
|
|
}
|
|
|
|
m_totalReceivedPackets++;
|
|
if (peer)
|
|
peer->m_totalPacketReceived++;
|
|
|
|
if (peer && (command->header.command & ENetProtocolFlag_Acknowledge) != 0)
|
|
{
|
|
UInt16 sentTime;
|
|
|
|
if (!(flags & ENetProtocolHeaderFlag_SentTime))
|
|
break;
|
|
|
|
sentTime = NetToHost(header->sentTime);
|
|
|
|
switch (peer->GetState())
|
|
{
|
|
case ENetPeerState::Disconnecting:
|
|
case ENetPeerState::AcknowledgingConnect:
|
|
case ENetPeerState::Disconnected:
|
|
case ENetPeerState::Zombie:
|
|
break;
|
|
|
|
case ENetPeerState::AcknowledgingDisconnect:
|
|
if ((command->header.command & ENetProtocolCommand_Mask) == ENetProtocolCommand_Disconnect)
|
|
peer->QueueAcknowledgement(command, sentTime);
|
|
break;
|
|
|
|
default:
|
|
peer->QueueAcknowledgement(command, sentTime);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return commandError();
|
|
}
|
|
|
|
int ENetHost::ReceiveIncomingCommands(ENetEvent* event)
|
|
{
|
|
for (unsigned int i = 0; i < 256; ++i)
|
|
{
|
|
bool shouldReceive = true;
|
|
std::size_t receivedLength;
|
|
|
|
if (m_isSimulationEnabled)
|
|
{
|
|
for (auto it = m_pendingIncomingPackets.begin(); it != m_pendingIncomingPackets.end(); ++it)
|
|
{
|
|
if (m_serviceTime >= it->deliveryTime)
|
|
{
|
|
shouldReceive = false;
|
|
|
|
m_receivedAddress = it->from;
|
|
receivedLength = it->data.GetDataSize();
|
|
std::memcpy(m_packetData[0].data(), it->data.GetConstData() + NetPacket::HeaderSize, receivedLength);
|
|
|
|
m_pendingIncomingPackets.erase(it);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (shouldReceive)
|
|
{
|
|
if (!m_socket.Receive(m_packetData[0].data(), m_packetData[0].size(), &m_receivedAddress, &receivedLength))
|
|
return -1; //< Error
|
|
|
|
if (receivedLength == 0)
|
|
return 0;
|
|
|
|
if (m_isSimulationEnabled)
|
|
{
|
|
if (m_packetLossProbability(s_randomGenerator))
|
|
continue;
|
|
|
|
UInt16 delay = m_packetDelayDistribution(s_randomGenerator);
|
|
if (delay > 0)
|
|
{
|
|
PendingIncomingPacket pendingPacket;
|
|
pendingPacket.deliveryTime = m_serviceTime + delay;
|
|
pendingPacket.from = m_receivedAddress;
|
|
pendingPacket.data.Reset(0, m_packetData[0].data(), receivedLength);
|
|
|
|
auto it = std::upper_bound(m_pendingIncomingPackets.begin(), m_pendingIncomingPackets.end(), pendingPacket, [] (const PendingIncomingPacket& first, const PendingIncomingPacket& second)
|
|
{
|
|
return first.deliveryTime < second.deliveryTime;
|
|
});
|
|
|
|
m_pendingIncomingPackets.emplace(it, std::move(pendingPacket));
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
m_receivedData = m_packetData[0].data();
|
|
m_receivedDataLength = receivedLength;
|
|
|
|
m_totalReceivedData += receivedLength;
|
|
|
|
// Intercept
|
|
|
|
if (HandleIncomingCommands(event))
|
|
return 1;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
void ENetHost::NotifyConnect(ENetPeer* peer, ENetEvent* event, bool incoming)
|
|
{
|
|
m_recalculateBandwidthLimits = true;
|
|
|
|
if (event)
|
|
{
|
|
peer->ChangeState(ENetPeerState::Connected);
|
|
|
|
event->type = (incoming) ? ENetEventType::IncomingConnect : ENetEventType::OutgoingConnect;
|
|
event->peer = peer;
|
|
event->data = peer->m_eventData;
|
|
}
|
|
else
|
|
peer->DispatchState((peer->GetState() == ENetPeerState::Connecting) ? ENetPeerState::ConnectionSucceeded : ENetPeerState::ConnectionPending);
|
|
}
|
|
|
|
void ENetHost::NotifyDisconnect(ENetPeer* peer, ENetEvent* event, bool timeout)
|
|
{
|
|
if (peer->GetState() >= ENetPeerState::ConnectionPending)
|
|
m_recalculateBandwidthLimits = true;
|
|
|
|
if (peer->GetState() != ENetPeerState::Connecting && (peer->GetState() < ENetPeerState::ConnectionSucceeded))
|
|
peer->Reset();
|
|
else if (event)
|
|
{
|
|
event->type = (timeout) ? ENetEventType::DisconnectTimeout : ENetEventType::Disconnect;
|
|
event->peer = peer;
|
|
event->data = peer->m_eventData;
|
|
|
|
peer->Reset();
|
|
}
|
|
else
|
|
{
|
|
peer->m_eventData = 0;
|
|
peer->m_timedOut = timeout;
|
|
|
|
peer->DispatchState(ENetPeerState::Zombie);
|
|
}
|
|
}
|
|
|
|
void ENetHost::SendAcknowledgements(ENetPeer* peer)
|
|
{
|
|
auto it = peer->m_acknowledgements.begin();
|
|
for (; it != peer->m_acknowledgements.end(); ++it)
|
|
{
|
|
if (m_commandCount >= m_commands.size() || m_bufferCount >= m_buffers.size() || peer->GetMtu() - m_packetSize < sizeof(ENetProtocolAcknowledge))
|
|
{
|
|
m_continueSending = true;
|
|
break;
|
|
}
|
|
|
|
ENetPeer::Acknowledgement& acknowledgement = *it;
|
|
|
|
ENetProtocol& command = m_commands[m_commandCount];
|
|
NetBuffer& buffer = m_buffers[m_bufferCount];
|
|
|
|
buffer.data = &command;
|
|
buffer.dataLength = sizeof(ENetProtocolAcknowledge);
|
|
|
|
m_packetSize += buffer.dataLength;
|
|
|
|
UInt16 reliableSequenceNumber = HostToNet(acknowledgement.command.header.reliableSequenceNumber);
|
|
|
|
command.header.command = ENetProtocolCommand_Acknowledge;
|
|
command.header.channelID = acknowledgement.command.header.channelID;
|
|
command.header.reliableSequenceNumber = reliableSequenceNumber;
|
|
command.acknowledge.receivedReliableSequenceNumber = reliableSequenceNumber;
|
|
command.acknowledge.receivedSentTime = HostToNet(UInt16(acknowledgement.sentTime));
|
|
|
|
if ((acknowledgement.command.header.command & ENetProtocolCommand_Mask) == ENetProtocolCommand_Disconnect)
|
|
peer->DispatchState(ENetPeerState::Zombie);
|
|
|
|
++m_bufferCount;
|
|
++m_commandCount;
|
|
}
|
|
|
|
peer->m_acknowledgements.erase(peer->m_acknowledgements.begin(), it);
|
|
}
|
|
|
|
bool ENetHost::SendReliableOutgoingCommands(ENetPeer* peer)
|
|
{
|
|
bool canPing = true;
|
|
bool windowExceeded = false;
|
|
bool windowWrap = false;
|
|
|
|
auto currentCommand = peer->m_outgoingReliableCommands.begin();
|
|
while (currentCommand != peer->m_outgoingReliableCommands.end())
|
|
{
|
|
auto outgoingCommand = currentCommand;
|
|
|
|
UInt16 reliableWindow = outgoingCommand->reliableSequenceNumber / ENetConstants::ENetPeer_ReliableWindowSize;
|
|
ENetPeer::Channel* channel = (outgoingCommand->command.header.channelID < peer->m_channels.size()) ? &peer->m_channels[outgoingCommand->command.header.channelID] : nullptr;
|
|
if (channel)
|
|
{
|
|
if (!windowWrap && outgoingCommand->sendAttempts < 1 && !(outgoingCommand->reliableSequenceNumber % ENetPeer_ReliableWindowSize) &&
|
|
((channel->reliableWindows[(reliableWindow + ENetPeer_ReliableWindows - 1) % ENetPeer_ReliableWindows] >= ENetPeer_ReliableWindowSize) ||
|
|
channel->usedReliableWindows & ((((1 << ENetPeer_ReliableWindows) - 1) << reliableWindow) |
|
|
(((1 << ENetPeer_FreeReliableWindows) - 1) >> (ENetPeer_ReliableWindows - reliableWindow)))))
|
|
windowWrap = true;
|
|
|
|
if (windowWrap)
|
|
{
|
|
++currentCommand;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (outgoingCommand->packet)
|
|
{
|
|
if (!windowExceeded)
|
|
{
|
|
UInt32 windowSize = (peer->m_packetThrottle * peer->m_windowSize) / ENetPeer_PacketThrottleScale;
|
|
|
|
if (peer->m_reliableDataInTransit + outgoingCommand->fragmentLength > std::max(windowSize, peer->GetMtu()))
|
|
windowExceeded = true;
|
|
}
|
|
|
|
if (windowExceeded)
|
|
{
|
|
++currentCommand;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
canPing = false;
|
|
|
|
assert((outgoingCommand->command.header.command & ENetProtocolCommand_Mask) < ENetProtocolCommand_Count);
|
|
std::size_t commandSize = s_enetCommandSizes[outgoingCommand->command.header.command & ENetProtocolCommand_Mask];
|
|
if (m_commandCount >= m_commands.size() || m_bufferCount + 1 >= m_buffers.size() || peer->GetMtu() - m_packetSize < commandSize ||
|
|
(outgoingCommand->packet && UInt16(peer->GetMtu() - m_packetSize) < UInt16(commandSize + outgoingCommand->fragmentLength)))
|
|
{
|
|
m_continueSending = true;
|
|
break;
|
|
}
|
|
|
|
++currentCommand;
|
|
|
|
if (channel && outgoingCommand->sendAttempts < 1)
|
|
{
|
|
channel->usedReliableWindows |= 1 << reliableWindow;
|
|
++channel->reliableWindows[reliableWindow];
|
|
}
|
|
|
|
++outgoingCommand->sendAttempts;
|
|
|
|
if (outgoingCommand->roundTripTimeout == 0)
|
|
{
|
|
outgoingCommand->roundTripTimeout = peer->m_roundTripTime + 4 * peer->m_roundTripTimeVariance;
|
|
outgoingCommand->roundTripTimeoutLimit = peer->m_timeoutLimit * outgoingCommand->roundTripTimeout;
|
|
}
|
|
|
|
if (peer->m_sentReliableCommands.empty())
|
|
peer->m_nextTimeout = m_serviceTime + outgoingCommand->roundTripTimeout;
|
|
|
|
peer->m_sentReliableCommands.emplace_back(std::move(*outgoingCommand));
|
|
|
|
peer->m_outgoingReliableCommands.erase(outgoingCommand);
|
|
outgoingCommand = peer->m_sentReliableCommands.end();
|
|
--outgoingCommand;
|
|
|
|
outgoingCommand->sentTime = m_serviceTime;
|
|
|
|
ENetProtocol& command = m_commands[m_commandCount];
|
|
NetBuffer& buffer = m_buffers[m_bufferCount];
|
|
|
|
buffer.data = &command;
|
|
buffer.dataLength = commandSize;
|
|
|
|
m_packetSize += buffer.dataLength;
|
|
m_headerFlags |= ENetProtocolHeaderFlag_SentTime;
|
|
|
|
command = outgoingCommand->command;
|
|
|
|
if (outgoingCommand->packet)
|
|
{
|
|
++m_bufferCount;
|
|
|
|
NetBuffer& packetBuffer = m_buffers[m_bufferCount];
|
|
packetBuffer.data = outgoingCommand->packet->data.GetData() + NetPacket::HeaderSize + outgoingCommand->fragmentOffset;
|
|
packetBuffer.dataLength = outgoingCommand->fragmentLength;
|
|
|
|
m_packetSize += packetBuffer.dataLength;
|
|
|
|
peer->m_reliableDataInTransit += outgoingCommand->fragmentLength;
|
|
}
|
|
|
|
++peer->m_packetsSent;
|
|
++peer->m_totalPacketSent;
|
|
++m_bufferCount;
|
|
++m_commandCount;
|
|
}
|
|
|
|
return canPing;
|
|
}
|
|
|
|
int ENetHost::SendOutgoingCommands(ENetEvent* event, bool checkForTimeouts)
|
|
{
|
|
std::array<UInt8, sizeof(ENetProtocolHeader) + sizeof(UInt32)> headerData;
|
|
ENetProtocolHeader* header = reinterpret_cast<ENetProtocolHeader*>(headerData.data());
|
|
|
|
m_continueSending = true;
|
|
|
|
while (m_continueSending)
|
|
{
|
|
m_continueSending = false;
|
|
|
|
for (std::size_t peer = 0; peer < m_peerCount; ++peer)
|
|
{
|
|
ENetPeer* currentPeer = &m_peers[peer];
|
|
if (currentPeer->GetState() == ENetPeerState::Disconnected || currentPeer->GetState() == ENetPeerState::Zombie)
|
|
continue;
|
|
|
|
m_headerFlags = 0;
|
|
m_commandCount = 0;
|
|
m_bufferCount = 1;
|
|
m_packetSize = sizeof(ENetProtocolHeader);
|
|
|
|
if (!currentPeer->m_acknowledgements.empty())
|
|
SendAcknowledgements(currentPeer);
|
|
|
|
if (checkForTimeouts && !currentPeer->m_sentReliableCommands.empty() && ENetTimeGreaterEqual(m_serviceTime, currentPeer->m_nextTimeout) && currentPeer->CheckTimeouts(event))
|
|
{
|
|
if (event && event->type != ENetEventType::None)
|
|
return 1;
|
|
else
|
|
continue;
|
|
}
|
|
|
|
if ((currentPeer->m_outgoingReliableCommands.empty() || SendReliableOutgoingCommands(currentPeer)) && currentPeer->m_sentReliableCommands.empty() &&
|
|
ENetTimeDifference(m_serviceTime, currentPeer->m_lastReceiveTime) >= currentPeer->m_pingInterval && currentPeer->m_mtu - m_packetSize >= sizeof(ENetProtocolPing))
|
|
{
|
|
currentPeer->Ping();
|
|
SendReliableOutgoingCommands(currentPeer);
|
|
}
|
|
|
|
if (!currentPeer->m_outgoingUnreliableCommands.empty())
|
|
SendUnreliableOutgoingCommands(currentPeer);
|
|
|
|
if (m_commandCount == 0)
|
|
continue;
|
|
|
|
if (currentPeer->m_packetLossEpoch == 0)
|
|
currentPeer->m_packetLossEpoch = m_serviceTime;
|
|
else if (ENetTimeDifference(m_serviceTime, currentPeer->m_packetLossEpoch) >= ENetPeer_PacketLossInterval && currentPeer->m_packetsSent > 0)
|
|
{
|
|
UInt32 packetLoss = currentPeer->m_packetsLost * ENetPeer_PacketLossScale / currentPeer->m_packetsSent;
|
|
|
|
#ifdef ENET_DEBUG
|
|
printf("peer %u: %f%%+-%f%% packet loss, %u+-%u ms round trip time, %f%% throttle, %u/%u outgoing, %u/%u incoming\n", currentPeer->incomingPeerID, currentPeer->packetLoss / (float) ENET_PEER_PACKET_LOSS_SCALE, currentPeer->packetLossVariance / (float) ENET_PEER_PACKET_LOSS_SCALE, currentPeer->roundTripTime, currentPeer->roundTripTimeVariance, currentPeer->packetThrottle / (float) ENET_PEER_PACKET_THROTTLE_SCALE, enet_list_size(¤tPeer->outgoingReliableCommands), enet_list_size(¤tPeer->outgoingUnreliableCommands), currentPeer->channels != NULL ? enet_list_size(¤tPeer->channels->incomingReliableCommands) : 0, currentPeer->channels != NULL ? enet_list_size(¤tPeer->channels->incomingUnreliableCommands) : 0);
|
|
#endif
|
|
|
|
currentPeer->m_packetLossVariance -= currentPeer->m_packetLossVariance / 4;
|
|
|
|
if (packetLoss >= currentPeer->m_packetLoss)
|
|
{
|
|
currentPeer->m_packetLoss += (packetLoss - currentPeer->m_packetLoss) / 8;
|
|
currentPeer->m_packetLossVariance += (packetLoss - currentPeer->m_packetLoss) / 4;
|
|
}
|
|
else
|
|
{
|
|
currentPeer->m_packetLoss -= (currentPeer->m_packetLoss - packetLoss) / 8;
|
|
currentPeer->m_packetLossVariance += (currentPeer->m_packetLoss - packetLoss) / 4;
|
|
}
|
|
|
|
currentPeer->m_packetLossEpoch = m_serviceTime;
|
|
currentPeer->m_packetsSent = 0;
|
|
currentPeer->m_packetsLost = 0;
|
|
}
|
|
|
|
|
|
m_buffers[0].data = headerData.data();
|
|
if (m_headerFlags & ENetProtocolHeaderFlag_SentTime)
|
|
{
|
|
header->sentTime = HostToNet(static_cast<UInt16>(m_serviceTime));
|
|
|
|
m_buffers[0].dataLength = sizeof(ENetProtocolHeader);
|
|
}
|
|
else
|
|
m_buffers[0].dataLength = NazaraOffsetOf(ENetProtocolHeader, sentTime);
|
|
|
|
// Compress packet buffers if possible
|
|
std::size_t compressedSize = 0;
|
|
if (m_compressor)
|
|
{
|
|
compressedSize = m_compressor->Compress(currentPeer, &m_buffers[1], m_bufferCount - 1, m_packetSize - sizeof(ENetProtocolHeader), m_packetData[1].data(), m_packetData[1].size());
|
|
if (compressedSize > 0)
|
|
m_headerFlags |= ENetProtocolHeaderFlag_Compressed;
|
|
}
|
|
|
|
if (currentPeer->m_outgoingPeerID < ENetConstants::ENetProtocol_MaximumPeerId)
|
|
m_headerFlags |= currentPeer->m_outgoingSessionID << ENetProtocolHeaderSessionShift;
|
|
|
|
header->peerID = HostToNet(static_cast<UInt16>(currentPeer->m_outgoingPeerID | m_headerFlags));
|
|
|
|
if (compressedSize > 0)
|
|
{
|
|
m_buffers[1].data = m_packetData[1].data();
|
|
m_buffers[1].dataLength = compressedSize;
|
|
m_bufferCount = 2;
|
|
}
|
|
|
|
currentPeer->m_lastSendTime = m_serviceTime;
|
|
|
|
// Simulate network by adding delay to packet sending and losing some packets
|
|
bool sendNow = true;
|
|
if (currentPeer->IsSimulationEnabled())
|
|
{
|
|
sendNow = false;
|
|
if (!currentPeer->m_packetLossProbability(s_randomGenerator))
|
|
{
|
|
UInt16 delay = currentPeer->m_packetDelayDistribution(s_randomGenerator);
|
|
if (delay == 0)
|
|
sendNow = true;
|
|
else
|
|
{
|
|
PendingOutgoingPacket outgoingPacket;
|
|
outgoingPacket.deliveryTime = m_serviceTime + delay;
|
|
outgoingPacket.to = currentPeer->GetAddress();
|
|
|
|
// Accumulate every temporary buffer into a datagram
|
|
for (std::size_t i = 0; i < m_bufferCount; ++i)
|
|
{
|
|
NetBuffer& buffer = m_buffers[i];
|
|
outgoingPacket.data.Write(buffer.data, buffer.dataLength);
|
|
}
|
|
|
|
m_totalSentData += outgoingPacket.data.GetDataSize();
|
|
|
|
// Add it to the right place
|
|
auto it = std::upper_bound(m_pendingOutgoingPackets.begin(), m_pendingOutgoingPackets.end(), outgoingPacket, [](const PendingOutgoingPacket& first, const PendingOutgoingPacket& second)
|
|
{
|
|
return first.deliveryTime < second.deliveryTime;
|
|
});
|
|
|
|
m_pendingOutgoingPackets.emplace(it, std::move(outgoingPacket));
|
|
}
|
|
}
|
|
}
|
|
|
|
if (sendNow)
|
|
{
|
|
std::size_t sentLength = 0;
|
|
|
|
if (!m_socket.SendMultiple(currentPeer->GetAddress(), m_buffers.data(), m_bufferCount, &sentLength))
|
|
{
|
|
switch (m_socket.GetLastError())
|
|
{
|
|
case SocketError::NetworkError:
|
|
case SocketError::UnreachableHost:
|
|
{
|
|
if (!currentPeer->IsConnected())
|
|
{
|
|
//< Network is down or unreachable (ex: IPv6 address when not supported), fails peer connection immediately
|
|
NotifyDisconnect(currentPeer, event, true);
|
|
return 1;
|
|
}
|
|
|
|
[[fallthrough]];
|
|
}
|
|
|
|
default:
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
m_totalSentData += sentLength;
|
|
}
|
|
|
|
currentPeer->RemoveSentUnreliableCommands();
|
|
m_totalSentPackets++;
|
|
}
|
|
}
|
|
|
|
if (!m_pendingOutgoingPackets.empty())
|
|
{
|
|
auto it = m_pendingOutgoingPackets.begin();
|
|
for (; it != m_pendingOutgoingPackets.end(); ++it)
|
|
{
|
|
if (m_serviceTime < it->deliveryTime)
|
|
break;
|
|
|
|
if (!m_socket.Send(it->to, it->data.GetConstData() + NetPacket::HeaderSize, it->data.GetDataSize(), nullptr))
|
|
return -1;
|
|
}
|
|
|
|
m_pendingOutgoingPackets.erase(m_pendingOutgoingPackets.begin(), it);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void ENetHost::SendUnreliableOutgoingCommands(ENetPeer* peer)
|
|
{
|
|
auto currentCommand = peer->m_outgoingUnreliableCommands.begin();
|
|
while (currentCommand != peer->m_outgoingUnreliableCommands.end())
|
|
{
|
|
auto outgoingCommand = currentCommand;
|
|
|
|
assert((outgoingCommand->command.header.command & ENetProtocolCommand_Mask) < ENetProtocolCommand_Count);
|
|
std::size_t commandSize = s_enetCommandSizes[outgoingCommand->command.header.command & ENetProtocolCommand_Mask];
|
|
|
|
if (m_commandCount >= m_commands.size() || m_bufferCount + 1 >= m_buffers.size() || peer->m_mtu - m_packetSize < commandSize ||
|
|
(outgoingCommand->packet && peer->m_mtu - m_packetSize < commandSize + outgoingCommand->fragmentLength))
|
|
{
|
|
m_continueSending = true;
|
|
break;
|
|
}
|
|
|
|
++currentCommand;
|
|
|
|
if (outgoingCommand->packet && outgoingCommand->fragmentOffset == 0)
|
|
{
|
|
peer->m_packetThrottleCounter += ENetConstants::ENetPeer_PacketThrottleCounter;
|
|
peer->m_packetThrottleCounter %= ENetConstants::ENetPeer_PacketThrottleScale;
|
|
|
|
if (peer->m_packetThrottleCounter > peer->m_packetThrottle)
|
|
{
|
|
UInt16 reliableSequenceNumber = outgoingCommand->reliableSequenceNumber;
|
|
UInt16 unreliableSequenceNumber = outgoingCommand->unreliableSequenceNumber;
|
|
|
|
for (;;)
|
|
{
|
|
peer->m_outgoingUnreliableCommands.erase(outgoingCommand);
|
|
|
|
if (currentCommand == peer->m_outgoingUnreliableCommands.end())
|
|
break;
|
|
|
|
outgoingCommand = currentCommand;
|
|
if (outgoingCommand->reliableSequenceNumber != reliableSequenceNumber || outgoingCommand->unreliableSequenceNumber != unreliableSequenceNumber)
|
|
break;
|
|
|
|
++currentCommand;
|
|
}
|
|
|
|
continue;
|
|
}
|
|
}
|
|
|
|
ENetProtocol& command = m_commands[m_commandCount];
|
|
NetBuffer& buffer = m_buffers[m_bufferCount];
|
|
|
|
buffer.data = &command;
|
|
buffer.dataLength = commandSize;
|
|
|
|
m_packetSize += buffer.dataLength;
|
|
|
|
command = outgoingCommand->command;
|
|
|
|
if (outgoingCommand->packet)
|
|
{
|
|
++m_bufferCount;
|
|
|
|
NetBuffer& packetBuffer = m_buffers[m_bufferCount];
|
|
packetBuffer.data = outgoingCommand->packet->data.GetData() + NetPacket::HeaderSize + outgoingCommand->fragmentOffset;
|
|
packetBuffer.dataLength = outgoingCommand->fragmentLength;
|
|
|
|
m_packetSize += packetBuffer.dataLength;
|
|
|
|
// In order to keep the packet buffer alive until we send it, place it into a temporary queue
|
|
peer->m_sentUnreliableCommands.emplace_back(std::move(*outgoingCommand));
|
|
}
|
|
|
|
peer->m_outgoingUnreliableCommands.erase(outgoingCommand);
|
|
|
|
++m_bufferCount;
|
|
++m_commandCount;
|
|
}
|
|
|
|
if (peer->GetState() == ENetPeerState::DisconnectLater && !peer->HasPendingCommands())
|
|
peer->Disconnect(peer->m_eventData);
|
|
}
|
|
|
|
void ENetHost::ThrottleBandwidth()
|
|
{
|
|
UInt32 currentTime = UInt32(GetElapsedMilliseconds().AsMilliseconds());
|
|
UInt32 elapsedTime = currentTime - m_bandwidthThrottleEpoch;
|
|
|
|
if (elapsedTime < ENetConstants::ENetHost_BandwidthThrottleInterval)
|
|
return;
|
|
|
|
m_bandwidthThrottleEpoch = currentTime;
|
|
|
|
if (m_connectedPeers == 0)
|
|
return;
|
|
|
|
UInt32 dataTotal = ~UInt32(0);
|
|
UInt32 bandwidth = ~UInt32(0);
|
|
|
|
if (m_outgoingBandwidth != 0)
|
|
{
|
|
bandwidth = (m_outgoingBandwidth * elapsedTime) / 1000;
|
|
|
|
dataTotal = 0;
|
|
for (ENetPeer& peer : m_peers)
|
|
{
|
|
if (peer.IsConnected())
|
|
continue;
|
|
|
|
dataTotal += peer.m_outgoingDataTotal;
|
|
}
|
|
}
|
|
|
|
UInt32 peersRemaining = m_connectedPeers;
|
|
UInt32 bandwidthLimit = ~UInt32(0);
|
|
UInt32 throttle = ~UInt32(0);
|
|
bool needsAdjustment = m_bandwidthLimitedPeers > 0;
|
|
|
|
while (peersRemaining > 0 && needsAdjustment)
|
|
{
|
|
needsAdjustment = false;
|
|
|
|
if (dataTotal <= bandwidth)
|
|
throttle = ENetConstants::ENetPeer_PacketThrottleScale;
|
|
else
|
|
throttle = (bandwidth * ENetConstants::ENetPeer_PacketThrottleScale) / dataTotal;
|
|
|
|
for (ENetPeer& peer : m_peers)
|
|
{
|
|
if (!peer.IsConnected() || peer.m_incomingBandwidth == 0 || peer.m_outgoingBandwidthThrottleEpoch == currentTime)
|
|
continue;
|
|
|
|
UInt32 peerBandwidth = (peer.m_incomingBandwidth * elapsedTime) / 1000;
|
|
if ((throttle * peer.m_outgoingDataTotal) / ENetConstants::ENetPeer_PacketThrottleScale <= peerBandwidth)
|
|
continue;
|
|
|
|
peer.m_packetThrottleLimit = Clamp<UInt32>((peerBandwidth * ENetConstants::ENetPeer_PacketThrottleScale) / peer.m_outgoingDataTotal, 0, peer.m_packetThrottleLimit);
|
|
peer.m_outgoingBandwidthThrottleEpoch = currentTime;
|
|
|
|
peer.m_incomingDataTotal = 0;
|
|
peer.m_outgoingDataTotal = 0;
|
|
|
|
needsAdjustment = true;
|
|
--peersRemaining;
|
|
bandwidth -= peerBandwidth;
|
|
dataTotal -= peerBandwidth;
|
|
}
|
|
}
|
|
|
|
if (peersRemaining > 0)
|
|
{
|
|
if (dataTotal <= bandwidth)
|
|
throttle = ENetConstants::ENetPeer_PacketThrottleScale;
|
|
else
|
|
throttle = (bandwidth * ENetConstants::ENetPeer_PacketThrottleScale) / dataTotal;
|
|
|
|
for (ENetPeer& peer : m_peers)
|
|
{
|
|
if (!peer.IsConnected() || peer.m_outgoingBandwidthThrottleEpoch == currentTime)
|
|
continue;
|
|
|
|
peer.m_packetThrottleLimit = throttle;
|
|
peer.m_packetThrottle = std::min(peer.m_packetThrottle, peer.m_packetThrottleLimit);
|
|
|
|
peer.m_incomingDataTotal = 0;
|
|
peer.m_outgoingDataTotal = 0;
|
|
}
|
|
}
|
|
|
|
if (m_recalculateBandwidthLimits)
|
|
{
|
|
m_recalculateBandwidthLimits = false;
|
|
|
|
peersRemaining = m_connectedPeers;
|
|
bandwidth = m_incomingBandwidth;
|
|
needsAdjustment = true;
|
|
|
|
if (bandwidth == 0)
|
|
bandwidthLimit = 0;
|
|
else
|
|
{
|
|
while (peersRemaining > 0 && needsAdjustment)
|
|
{
|
|
needsAdjustment = false;
|
|
bandwidthLimit = bandwidth / peersRemaining;
|
|
|
|
for (ENetPeer& peer : m_peers)
|
|
{
|
|
if (!peer.IsConnected() || peer.m_incomingBandwidthThrottleEpoch == currentTime)
|
|
continue;
|
|
|
|
if (peer.m_outgoingBandwidth > 0 && peer.m_outgoingBandwidth >= bandwidthLimit)
|
|
continue;
|
|
|
|
peer.m_incomingBandwidthThrottleEpoch = currentTime;
|
|
|
|
needsAdjustment = true;
|
|
--peersRemaining;
|
|
bandwidth -= peer.m_outgoingBandwidth;
|
|
}
|
|
}
|
|
}
|
|
|
|
for (ENetPeer& peer : m_peers)
|
|
{
|
|
if (!peer.IsConnected())
|
|
continue;
|
|
|
|
ENetProtocol command(ENetProtocolCommand_BandwidthLimit | ENetProtocolFlag_Acknowledge, 0xFF);
|
|
command.bandwidthLimit.outgoingBandwidth = HostToNet(m_outgoingBandwidth);
|
|
|
|
if (peer.m_incomingBandwidthThrottleEpoch == currentTime)
|
|
command.bandwidthLimit.incomingBandwidth = HostToNet(peer.m_outgoingBandwidth);
|
|
else
|
|
command.bandwidthLimit.incomingBandwidth = HostToNet(bandwidthLimit);
|
|
|
|
peer.QueueOutgoingCommand(command);
|
|
}
|
|
}
|
|
}
|
|
|
|
std::size_t ENetHost::GetCommandSize(UInt8 commandNumber)
|
|
{
|
|
assert((commandNumber & ENetProtocolCommand_Mask) < ENetProtocolCommand_Count);
|
|
return s_enetCommandSizes[commandNumber & ENetProtocolCommand_Mask];
|
|
}
|
|
|
|
bool ENetHost::Initialize()
|
|
{
|
|
std::random_device device;
|
|
s_randomGenerator.seed(device());
|
|
s_randomGenerator64.seed(device());
|
|
|
|
return true;
|
|
}
|
|
|
|
void ENetHost::Uninitialize()
|
|
{
|
|
}
|
|
|
|
std::mt19937 ENetHost::s_randomGenerator;
|
|
std::mt19937_64 ENetHost::s_randomGenerator64;
|
|
}
|