mirror of
https://github.com/BeamMP/BeamMP-Server.git
synced 2026-07-12 17:54:04 +00:00
accept without EP, explicitly (and fallibly) get the endpoint
This commit is contained in:
+19
-15
@@ -245,23 +245,27 @@ void TNetwork::TCPServerMain() {
|
|||||||
beammp_debug("shutdown during TCP wait for accept loop");
|
beammp_debug("shutdown during TCP wait for accept loop");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
boost::asio::ip::tcp::endpoint ClientEp;
|
boost::asio::ip::tcp::socket ClientSocket = Acceptor.accept(ec);
|
||||||
boost::asio::ip::tcp::socket ClientSocket = Acceptor.accept(ClientEp, ec);
|
if (ec) {
|
||||||
std::string ClientIP = ClientEp.address().to_string();
|
|
||||||
if (!ec) {
|
|
||||||
auto MaybeGuard = mConnectionLimiter.TryAcquire(ClientEp.address().to_v6().to_string());
|
|
||||||
if (MaybeGuard.has_value()) {
|
|
||||||
// move-swap to avoid copy ctor (deleted)
|
|
||||||
auto Guard = std::move(MaybeGuard.value());
|
|
||||||
TConnection Conn { std::move(ClientSocket), ClientEp };
|
|
||||||
std::thread ID(&TNetwork::Identify, this, std::move(Conn), std::move(Guard));
|
|
||||||
ID.detach(); // TODO: Add to a queue and attempt to join periodically
|
|
||||||
} else {
|
|
||||||
beammp_errorf("Connection rejected for {} due to the global or concurrent connection limit", ClientIP);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
beammp_errorf("Failed to accept() new client: {}", ec.message());
|
beammp_errorf("Failed to accept() new client: {}", ec.message());
|
||||||
|
continue;
|
||||||
}
|
}
|
||||||
|
boost::asio::ip::tcp::endpoint ClientEp = ClientSocket.remote_endpoint(ec);
|
||||||
|
if (ec) {
|
||||||
|
beammp_errorf("Accepted socket but failed to query remote endpoint for IP address: {}", ec.message());
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
std::string ClientIP = ClientEp.address().to_string();
|
||||||
|
auto MaybeGuard = mConnectionLimiter.TryAcquire(ClientIP);
|
||||||
|
if (!MaybeGuard.has_value()) {
|
||||||
|
beammp_errorf("Connection rejected for {} due to the global or concurrent connection limit", ClientIP);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
// move-swap to avoid copy ctor (deleted)
|
||||||
|
auto Guard = std::move(MaybeGuard.value());
|
||||||
|
TConnection Conn { std::move(ClientSocket), ClientEp };
|
||||||
|
std::thread ID(&TNetwork::Identify, this, std::move(Conn), std::move(Guard));
|
||||||
|
ID.detach(); // TODO: Add to a queue and attempt to join periodically
|
||||||
} catch (const std::exception& e) {
|
} catch (const std::exception& e) {
|
||||||
beammp_errorf("Exception in accept routine: {}", e.what());
|
beammp_errorf("Exception in accept routine: {}", e.what());
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user