timeline refactor, bit broken rn

This commit is contained in:
2025-11-09 10:35:21 -05:00
parent 1e35910b0a
commit e2799b1e58
41 changed files with 2034 additions and 1374 deletions

161
src/socket.cpp Normal file
View File

@@ -0,0 +1,161 @@
#include "socket.h"
namespace anm2ed
{
#ifdef _WIN32
namespace
{
struct WSAInitializer
{
WSAInitializer()
{
WSADATA data{};
WSAStartup(MAKEWORD(2, 2), &data);
}
~WSAInitializer() { WSACleanup(); }
};
WSAInitializer initializer{};
}
#endif
Socket::Socket() : handle(SOCKET_INVALID), role(CLIENT) {}
Socket::Socket(Socket&& other) noexcept : handle(other.handle), role(other.role) { other.handle = SOCKET_INVALID; }
Socket& Socket::operator=(Socket&& other) noexcept
{
if (this != &other)
{
close();
handle = other.handle;
role = other.role;
other.handle = SOCKET_INVALID;
}
return *this;
}
Socket::~Socket() { close(); }
bool Socket::open(SocketRole newRole)
{
close();
role = newRole;
handle = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (!is_valid()) return false;
if (role == SERVER)
{
#ifdef _WIN32
BOOL opt = TRUE;
#else
int opt = 1;
#endif
::setsockopt(handle, SOL_SOCKET, SO_REUSEADDR, reinterpret_cast<const char*>(&opt), sizeof(opt));
}
return true;
}
bool Socket::bind(const SocketAddress& address)
{
if (!is_valid()) return false;
sockaddr_in addr{};
addr.sin_family = AF_INET;
addr.sin_port = htons(address.port);
if (address.host.empty())
addr.sin_addr.s_addr = htonl(INADDR_ANY);
else
{
if (::inet_pton(AF_INET, address.host.c_str(), &addr.sin_addr) <= 0) return false;
}
return ::bind(handle, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) == 0;
}
bool Socket::listen()
{
if (!is_valid()) return false;
return ::listen(handle, SOMAXCONN) == 0;
}
Socket Socket::accept()
{
Socket client{};
if (!is_valid()) return client;
auto accepted = ::accept(handle, nullptr, nullptr);
if (accepted == SOCKET_INVALID) return client;
client.close();
client.handle = accepted;
client.role = CLIENT;
return client;
}
bool Socket::connect(const SocketAddress& address)
{
if (!is_valid()) return false;
sockaddr_in addr{};
addr.sin_family = AF_INET;
addr.sin_port = htons(address.port);
if (::inet_pton(AF_INET, address.host.c_str(), &addr.sin_addr) <= 0) return false;
return ::connect(handle, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) == 0;
}
bool Socket::send(const void* data, size_t size)
{
if (!is_valid() || !data || size == 0) return false;
auto bytes = reinterpret_cast<const char*>(data);
size_t totalSent = 0;
while (totalSent < size)
{
auto sent = ::send(handle, bytes + totalSent, static_cast<int>(size - totalSent), 0);
if (sent <= 0) return false;
totalSent += static_cast<size_t>(sent);
}
return true;
}
bool Socket::receive(void* buffer, size_t size)
{
if (!is_valid() || !buffer || size == 0) return false;
auto* bytes = reinterpret_cast<char*>(buffer);
size_t totalReceived = 0;
while (totalReceived < size)
{
auto received = ::recv(handle, bytes + totalReceived, static_cast<int>(size - totalReceived), 0);
if (received <= 0) return false;
totalReceived += static_cast<size_t>(received);
}
return true;
}
void Socket::close()
{
if (!is_valid()) return;
#ifdef _WIN32
::closesocket(handle);
#else
::close(handle);
#endif
handle = SOCKET_INVALID;
}
bool Socket::is_valid() const { return handle != SOCKET_INVALID; }
}