From f1a152c02b34979a27316a024f2f1450b9d1c278 Mon Sep 17 00:00:00 2001 From: littleblack111 Date: Sat, 31 May 2025 16:09:30 +0800 Subject: internal: all CFileDescriptor is now wrapped in shared_ptr shared_ptr operations are atomic good for multi threading --- src/server.cpp | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) (limited to 'src/server.cpp') diff --git a/src/server.cpp b/src/server.cpp index 9b5dae0..db92481 100644 --- a/src/server.cpp +++ b/src/server.cpp @@ -1,27 +1,28 @@ #include "server.hpp" #include "log.hpp" #include +#include #include CServer::CServer(uint16_t port) - : m_sockfd(CFileDescriptor{socket(AF_INET, SOCK_STREAM, 0)}) { - if (!m_sockfd.isValid()) + : m_sockfd(std::make_shared(socket(AF_INET, SOCK_STREAM, 0))) { + if (!m_sockfd->isValid()) throw std::runtime_error("Failed to create socket"); constexpr int opt = 1; constexpr int size = sizeof(opt); - if (setsockopt(m_sockfd.get(), SOL_SOCKET, SO_REUSEADDR, &opt, size) < 0) + if (setsockopt(m_sockfd->get(), SOL_SOCKET, SO_REUSEADDR, &opt, size) < 0) log(stderr, WARN, "setsockopt(SO_REUSEADDR) failed"); - if (setsockopt(m_sockfd.get(), SOL_SOCKET, SO_KEEPALIVE, &opt, size) < 0) + if (setsockopt(m_sockfd->get(), SOL_SOCKET, SO_KEEPALIVE, &opt, size) < 0) log(stderr, WARN, "setsockopt(SO_KEEPALIVE) failed"); memset(&m_addr, 0, sizeof(m_addr)); m_addr.sin_family = AF_INET; m_addr.sin_port = htons(port); m_addr.sin_addr.s_addr = INADDR_ANY; - bind(m_sockfd.get(), reinterpret_cast(&m_addr), sizeof(m_addr)) || - listen(m_sockfd.get(), 5) + bind(m_sockfd->get(), reinterpret_cast(&m_addr), sizeof(m_addr)) || + listen(m_sockfd->get(), 5) ? throw std::runtime_error("Failed to bind or listen") : log(LOG, "Server started on port: {}", port); @@ -33,6 +34,6 @@ CServer::~CServer() { log(SYS, "Server: bye"); } -const CFileDescriptor *CServer::getSocket() const { - return &m_sockfd; +std::shared_ptr CServer::getSocket() const { + return m_sockfd; } -- cgit v1.2.3