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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#include "libsock/server/server.hpp"
#include "libsock/types.hpp"
#include "misc/FileDescriptor.hpp"
#include <algorithm>
#include <cstring>
#include <mutex>
#include <stdexcept>
#include <sys/socket.h>
using namespace LibSock::Server;
namespace LibSock {
namespace Server {
Server::Server(uint16_t port, bool reuseaddr, bool keepalive)
: m_sockfd(std::make_shared<LibSock::CFileDescriptor>(socket(AF_INET, SOCK_STREAM, 0)))
, m_port(port) {
std::lock_guard<std::mutex> lk(m_mutex);
if (!m_sockfd->isValid() || m_sockfd->get() < 0)
throw std::runtime_error("server:server 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("server:server setsockopt(SO_REUSEADDR) failed");
if (reuseaddr && setsockopt(m_sockfd->get(), SOL_SOCKET, SO_KEEPALIVE, &opt, size) < 0)
throw std::runtime_error("server:server setsockopt(SO_KEEPALIVE) failed");
memset(&m_addr, 0, sizeof(m_addr));
m_addr.sin_family = AF_INET;
m_addr.sin_port = htons(m_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("server:server Failed to bind or listen on socket");
}
SP<Server> Server::create(uint16_t port, bool reuseaddr, bool keepalive) {
auto c = SP<Server>(new Server(port, reuseaddr, keepalive));
vpServer.emplace_back(c);
c->m_self = c;
return c;
}
WP<Server> Server::get() {
std::lock_guard<std::mutex> lk(m_mutex);
if (m_self.expired())
m_self = shared_from_this();
return m_self;
}
Server::~Server() {
m_sockfd.reset();
vpServer.erase(std::remove_if(vpServer.begin(), vpServer.end(), [this](const SP<Server> &sptr) { return sptr.get() == this; }), vpServer.end());
}
SP<LibSock::CFileDescriptor> Server::getSocket() const {
return m_sockfd;
}
} // namespace Server
} // namespace LibSock
|
