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
|
#include "libsock/server/client.hpp"
#include "libsock/server/clients.hpp"
#include "libsock/server/server.hpp"
#include "libsock/types.hpp"
#include "misc/FileDescriptor.hpp"
#include <algorithm>
#include <arpa/inet.h>
#include <cstring>
#include <format>
#include <memory>
#include <optional>
#include <sys/socket.h>
using namespace LibSock::Server;
Client::Client(SP<Server> server, SP<Clients> clients, bool track, bool oneShot)
: m_track(track)
, m_oneShot(oneShot) {
m_wpServer = std::move(server);
m_wpClients = std::move(clients);
if (m_wpClients.expired())
throw std::runtime_error("server:client: ClientManager doesn't exist");
if (m_wpServer.expired())
throw std::runtime_error("server:client: Server doesn't exist");
}
void Client::init() {
m_sockfd = std::make_shared<LibSock::CFileDescriptor>(accept(m_wpServer.lock()->getSocket()->get(), reinterpret_cast<sockaddr *>(&m_addr), &m_addrLen)); // if this is in the init list, it will run before
// m_addrLen, so it won't work :/
if (!m_sockfd->isValid())
throw std::runtime_error("server:client: Failed to create socket");
m_ip.resize(INET_ADDRSTRLEN);
inet_ntop(AF_INET, &m_addr.sin_addr, &m_ip[0], INET_ADDRSTRLEN);
m_ip.resize(strlen(m_ip.c_str()));
m_port = ntohs(m_addr.sin_port);
}
Client::~Client() {
m_sockfd.reset();
}
void Client::recvLoop(std::optional<std::function<bool(const SRecvData &)>> cb) {
while (true) {
auto recvData = read();
if (!recvData->good)
break;
recvData->sanitize();
if (recvData->isEmpty())
continue;
if (cb && !(*cb)(*recvData))
return;
}
}
bool SRecvData::isEmpty() const {
return data.empty() || std::ranges::all_of(data, [](char c) { return std::isspace(c); });
}
void SRecvData::sanitize() {
if (!data.empty() && data.back() == '\n')
data.pop_back();
data.erase(std::remove(data.begin(), data.end(), asciiEscape), data.end()); // don't accept ASCII code, might mess up terminal
// // NOLINT
if (const auto start = data.find_first_not_of(" \t\r\n");
start != std::string::npos) {
const auto end = data.find_last_not_of(" \t\r\n");
data = data.substr(start, end - start + 1);
} else
data.clear();
}
UP<SRecvData> Client::read(std::optional<std::function<bool(const SRecvData &)>> cb) {
auto recvData = std::make_unique<SRecvData>();
recvData->data.resize(recvData->size);
ssize_t size = recv(m_sockfd->get(), &recvData->data[0], recvData->size, 0);
m_szReading.reset();
if (size < 0 || size == 0) {
recvData->good = false;
if (cb && !(*cb)(*recvData))
return nullptr;
} else {
recvData->data.resize(size);
recvData->good = true;
}
if (m_track)
m_wpClients.lock()->m_vDatas.push_back({recvData->data, m_self});
return recvData;
}
UP<SRecvData> Client::read(const std::string &msg, std::optional<std::function<bool(const SRecvData &)>> cb) {
write("{}", msg);
m_szReading = msg;
return read(cb);
}
void Client::runLoop(bool resumeHist, std::optional<std::function<bool(const SRecvData &)>> cb) {
if (resumeHist)
resumeHistory();
recvLoop(cb);
m_wpClients.lock()->kick(m_self);
}
void Client::resumeHistory() {
for (const auto &chat : m_wpClients.lock()->getDatas())
write(chat.msg);
}
bool Client::isValid() {
if (!m_sockfd)
return false;
if (!m_sockfd->isValid())
return false;
int err = 0;
socklen_t size = sizeof(err);
return getsockopt(m_sockfd->get(), SOL_SOCKET, SO_ERROR, &err, &size) >= 0 && err == 0;
}
bool Client::write(const std::string &msg, std::optional<std::function<bool(const SRecvData &)>> cb) {
auto r = write("{}", msg);
if (cb)
r = (*cb)({.data = msg, .size = msg.size(), .good = r});
return r;
}
const std::string &Client::getIp() const { return m_ip; }
template <typename... Args>
bool Client::write(std::format_string<Args...> fmt, Args &&...args) {
std::string msg = std::format(fmt, std::forward<Args>(args)...);
if (m_szReading) {
msg.insert(0, "\r");
msg.append(*m_szReading);
}
if (send(m_sockfd->get(), msg.c_str(), msg.size(), 0) < 0) {
// cb somehow
return false;
}
return true;
}
|
