1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#include "libsock/server/server.hpp"
#include "misc/memory.hpp"
#include <cstring>
#include <mutex>
#include <sys/socket.h>
using namespace LibSock::Server;
Server::Server(uint16_t port, bool reuseaddr, bool keepalive)
: m_sockfd(
std::make_shared<CFileDescriptor>(socket(AF_INET, SOCK_STREAM, 0))) {
std::lock_guard<std::mutex> lk(m_mutex);
if (!m_sockfd->isValid() || m_sockfd->get() < 0)
throw std::runtime_error("Failed to create socket");
constexpr int opt = 1;
constexpr int size = sizeof(opt);
if (keepalive &&
setsockopt(m_sockfd->get(), SOL_SOCKET, SO_REUSEADDR, &opt, size) < 0)
throw std::runtime_error("setsockopt(SO_REUSEADDR) failed");
if (reuseaddr &&
setsockopt(m_sockfd->get(), SOL_SOCKET, SO_KEEPALIVE, &opt, size) < 0)
throw std::runtime_error("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;
if (bind(m_sockfd->get(), reinterpret_cast<sockaddr *>(&m_addr),
sizeof(m_addr)) ||
listen(m_sockfd->get(), 5))
throw std::runtime_error("Failed to bind or listen on socket");
};
Server::~Server() { m_sockfd.reset(); }
std::shared_ptr<CFileDescriptor> Server::getSocket() const { return m_sockfd; }
|
