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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
/*
* File: net/unique_socket.hpp
* Purpose: RAII socket concept in modern C++
* Author: Amlal El Mahrouss (amlal@nekernel.org)
* Copyright 2025, Amlal El Mahrouss, licensed under the Boost Software License.
*/
#pragma once
#include <core/config.hpp>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <string>
#include <cstddef>
#include <cstring>
#ifdef _WIN32
#error !!! "Windows is not supported yet for <socket>" !!!
#endif // _WIN32
namespace ocl::net
{
class unique_socket;
/// =============================================================================
/// @brief Modem container concept, a container to read and write on a network stream.
/// =============================================================================
class unique_socket final
{
public:
using socket_type = int64_t;
using error_type = bool;
using condition_type = bool;
private:
socket_type fd_{};
condition_type is_server_{false};
error_type bad_{true};
public:
const error_type& bad{bad_};
unique_socket() = default;
~unique_socket()
{
this->destroy();
}
unique_socket& operator=(const unique_socket&) = delete;
unique_socket(const unique_socket&) = default;
static constexpr auto local_address_ip4 = "127.0.0.1";
static constexpr auto local_address_ip6 = "::1";
static constexpr const auto backlog_count = 5U;
unique_socket read(const char* out, std::size_t len) noexcept
{
if (!out || !len)
return {};
socket_type cl_{fd_};
if (this->is_server_)
cl_ = ::accept(fd_, nullptr, nullptr);
auto ret = ::recv(cl_, static_cast<void*>(const_cast<char*>(out)), len, 0);
unique_socket sock;
sock.fd_ = cl_;
sock.bad_ = ret > 0L;
return std::move(sock);
}
unique_socket& write(const char* out, std::size_t len) noexcept
{
if (!out)
return *this;
if (!len)
return *this;
auto ret = ::send(fd_, out, len, 0);
bad_ = !(ret >= 0L);
return *this;
}
unique_socket& write(const std::string& out) noexcept
{
if (out.empty())
return *this;
auto ret = ::send(fd_, out.data(), out.size(), 0);
bad_ = !(ret >= 0L);
return *this;
}
template <uint16_t port>
static unique_socket make_socket(const std::string& address, const bool is_server)
{
unique_socket sock;
sock.construct<AF_INET, SOCK_STREAM, port>(address.c_str(), is_server);
return sock;
}
private:
template <uint16_t af, uint16_t kind, uint16_t port>
bool construct(const char* addr = unique_socket::local_address_ip4, const bool& is_server = false) noexcept
{
static_assert(af != 0, "Address family is zero");
static_assert(kind != 0, "Kind is zero");
fd_ = ::socket(af, kind, 0);
is_server_ = is_server;
if (fd_ == -1)
return false;
struct sockaddr_in addr_;
std::memset(&addr_, 0, sizeof(struct sockaddr_in));
addr_.sin_addr.s_addr = ::inet_addr(addr);
addr_.sin_port = htons(port);
addr_.sin_family = af;
if (!is_server)
{
const auto ret = ::connect(fd_, reinterpret_cast<struct sockaddr*>(&addr_), sizeof(addr_));
return ret == 0L;
}
int ret = ::bind(fd_, (struct sockaddr*)&addr_, sizeof(addr_));
bad_ = ret == -1;
::listen(fd_, unique_socket::backlog_count);
return bad_ == false;
}
bool destroy() noexcept
{
if (!fd_)
return false;
::shutdown(fd_, SHUT_RDWR);
::close(fd_);
fd_ = 0L;
return true;
}
};
} // namespace ocl::net
|
