Prepare TCP support

This commit is contained in:
momo5502
2025-01-11 21:29:55 +01:00
parent c8c1e000a3
commit 21e2f6f999
4 changed files with 133 additions and 0 deletions

View File

@@ -0,0 +1,40 @@
#include "tcp_client_socket.hpp"
namespace network
{
tcp_client_socket::tcp_client_socket(SOCKET s, const address& target)
: socket(s),
target_(target)
{
}
bool tcp_client_socket::send(const void* data, const size_t size) const
{
const auto res = ::send(this->get_socket(), static_cast<const char*>(data), static_cast<send_size>(size), 0);
return static_cast<size_t>(res) == size;
}
bool tcp_client_socket::send(const std::string_view data) const
{
return this->send(data.data(), data.size());
}
bool tcp_client_socket::receive(std::string& data) const
{
char buffer[0x2000];
const auto result = recv(this->get_socket(), buffer, static_cast<int>(sizeof(buffer)), 0);
if (result == SOCKET_ERROR)
{
return false;
}
data.assign(buffer, buffer + result);
return true;
}
address tcp_client_socket::get_target() const
{
return this->target_;
}
}

View File

@@ -0,0 +1,33 @@
#pragma once
#include "socket.hpp"
#include <string_view>
namespace network
{
class tcp_server_socket;
class tcp_client_socket : public socket
{
// TODO: Construct and connect client!
tcp_client_socket() = default;
~tcp_client_socket() = default;
tcp_client_socket(tcp_client_socket&& obj) noexcept = default;
tcp_client_socket& operator=(tcp_client_socket&& obj) noexcept = default;
[[maybe_unused]] bool send(const void* data, size_t size) const;
[[maybe_unused]] bool send(std::string_view data) const;
bool receive(std::string& data) const;
address get_target() const;
private:
address target_{};
friend tcp_server_socket;
tcp_client_socket(SOCKET s, const address& target);
};
}

View File

@@ -0,0 +1,35 @@
#include "tcp_server_socket.hpp"
namespace network
{
tcp_server_socket::tcp_server_socket(const int af)
: socket(af, SOCK_STREAM, IPPROTO_TCP)
{
}
tcp_client_socket tcp_server_socket::accept()
{
this->listen();
address a{};
auto len = a.get_max_size();
auto s = ::accept(this->get_socket(), &a.get_addr(), &len);
if (s == INVALID_SOCKET)
{
return {};
}
return tcp_client_socket{s, a};
}
void tcp_server_socket::listen()
{
if (this->listening_)
{
return;
}
this->listening_ = ::listen(this->get_socket(), 32) == 0;
}
}

View File

@@ -0,0 +1,25 @@
#pragma once
#include "socket.hpp"
#include "tcp_client_socket.hpp"
namespace network
{
class tcp_server_socket : public socket
{
tcp_server_socket(int af);
tcp_server_socket() = default;
~tcp_server_socket() = default;
tcp_server_socket(tcp_server_socket&& obj) noexcept = default;
tcp_server_socket& operator=(tcp_server_socket&& obj) noexcept = default;
tcp_client_socket accept();
private:
bool listening_{false};
void listen();
};
}