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
63
64
65
66
67
|
#include "libsock/server/server.hpp"
#include "libsock/types.hpp"
#include "misc/FileDescriptor.hpp"
#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) {
if (pServer)
throw std::runtime_error("server:server Server already exist");
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) {
if (pServer)
throw std::runtime_error("server:server Server already exist");
pServer = SP<Server>(new Server(port, reuseaddr, keepalive));
return pServer;
}
SP<Server> Server::get() {
if (!pServer)
pServer = shared_from_this();
return pServer;
}
Server::~Server() {
m_sockfd.reset();
}
SP<LibSock::CFileDescriptor> Server::getSocket() const {
return m_sockfd;
}
} // namespace Server
} // namespace LibSock
|
