voxelcoreprojects/voxelcore
1
0
Fork 0
mirror of https://github.com/MihailRis/voxelcore.git synced 2026-10-04 18:41:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make socket non-blocking

Browse source
This commit is contained in:
MihailRis 2024-11-26 17:12:24 +03:00
parent e8b674ca65
commit 18bdce52df
3 changed files with 37 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/network/Network.cpp
View file

@ -182,6 +182,7 @@ public:
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>
#endif
#ifndef _WIN32
@ -309,6 +310,22 @@ public:
freeaddrinfo(addrinfo);
throw std::runtime_error("Could not create socket");
}
#ifdef _WIN32
u_long mode = 1;
auto result = ioctlsocket(descriptor, FIONBIO, &mode);
if (result != NO_ERROR) {
throw std::runtime_error(
"Could not set to non-blocking mode [errno=" + std::to_string(err) +
"]: " + std::string(strerror(err))
);
}
#else
if (fcntl(descriptor, F_SETFL, O_NONBLOCK) < 0) {
freeaddrinfo(addrinfo);
closesocket(descriptor);
throw std::runtime_error("Failed to make socket non-blocking");
}
#endif
int res = connectsocket(descriptor, addrinfo->ai_addr, addrinfo->ai_addrlen);
if (res == -1) {
closesocket(descriptor);
@ -341,15 +358,24 @@ void Network::get(
requests->get(url, onResponse, onReject, maxSize);
}
std::shared_ptr<Socket> Network::connect(const std::string& address, int port) {
Socket* Network::getConnection(u64id_t id) const {
const auto& found = connections.find(id);
if (found == connections.end()) {
return nullptr;
}
return found->second.get();
}
u64id_t Network::connect(const std::string& address, int port) {
auto socket = SocketImpl::connect(address, port);
connections.push_back(socket);
return socket;
u64id_t id = nextConnection++;
connections[id] = std::move(socket);
return id;
}
size_t Network::getTotalUpload() const {
size_t totalUpload = 0;
for (const auto& socket : connections) {
for (const auto& [_, socket] : connections) {
totalUpload += socket->getTotalUpload();
}
return requests->getTotalUpload() + totalUpload;
@ -357,7 +383,7 @@ size_t Network::getTotalUpload() const {
size_t Network::getTotalDownload() const {
size_t totalDownload = 0;
for (const auto& socket : connections) {
for (const auto& [_, socket] : connections) {
totalDownload += socket->getTotalDownload();
}
return requests->getTotalDownload() + totalDownload;