blob: e790e60ae6a3bbc298e7da118e1a93d3e9af6045 (
plain)
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
|
#include <network/poller.hpp>
#include <assert.h>
#include <stdio.h>
#include <cstring>
#include <iostream>
Poller::Poller()
{
std::cout << "Poller()" << std::endl;
#if POLLER == POLL
memset(this->fds, 0, sizeof(this->fds));
this->nfds = 0;
#endif
}
Poller::~Poller()
{
std::cout << "~Poller()" << std::endl;
}
void Poller::add_socket_handler(std::shared_ptr<SocketHandler> socket_handler)
{
// Raise an error if that socket is already in the list
const auto it = this->socket_handlers.find(socket_handler->get_socket());
if (it != this->socket_handlers.end())
throw std::runtime_error("Trying to insert SocketHandler already managed");
this->socket_handlers.emplace(socket_handler->get_socket(), socket_handler);
socket_handler->set_poller(this);
// We always watch all sockets for receive events
#if POLLER == POLL
this->fds[this->nfds].fd = socket_handler->get_socket();
this->fds[this->nfds].events = POLLIN;
this->nfds++;
#endif
}
void Poller::remove_socket_handler(const socket_t socket)
{
const auto it = this->socket_handlers.find(socket);
if (it == this->socket_handlers.end())
throw std::runtime_error("Trying to remove a SocketHandler that is not managed");
this->socket_handlers.erase(it);
for (size_t i = 0; i < this->nfds; i++)
{
if (this->fds[i].fd == socket)
{
// Move all subsequent pollfd by one on the left, erasing the
// value of the one we remove
for (size_t j = i; j < this->nfds - 1; ++j)
{
this->fds[j].fd = this->fds[j+1].fd;
this->fds[j].events= this->fds[j+1].events;
}
this->nfds--;
}
}
}
void Poller::watch_send_events(const SocketHandler* const socket_handler)
{
#if POLLER == POLL
for (size_t i = 0; i <= this->nfds; ++i)
{
if (this->fds[i].fd == socket_handler->get_socket())
{
this->fds[i].events = POLLIN|POLLOUT;
return;
}
}
#endif
throw std::runtime_error("Cannot watch a non-registered socket for send events");
}
void Poller::stop_watching_send_events(const SocketHandler* const socket_handler)
{
#if POLLER == POLL
for (size_t i = 0; i <= this->nfds; ++i)
{
if (this->fds[i].fd == socket_handler->get_socket())
{
this->fds[i].events = POLLIN;
return;
}
}
#endif
throw std::runtime_error("Cannot watch a non-registered socket for send events");
}
bool Poller::poll()
{
#if POLLER == POLL
if (this->nfds == 0)
return false;
int res = ::poll(this->fds, this->nfds, -1);
if (res < 0)
{
perror("poll");
throw std::runtime_error("Poll failed");
}
// We cannot possibly have more ready events than the number of fds we are
// watching
assert(static_cast<unsigned int>(res) <= this->nfds);
for (size_t i = 0; i <= this->nfds && res != 0; ++i)
{
if (this->fds[i].revents == 0)
continue;
else if (this->fds[i].revents & POLLIN)
{
auto socket_handler = this->socket_handlers.at(this->fds[i].fd);
socket_handler->on_recv();
res--;
}
else if (this->fds[i].revents & POLLOUT)
{
auto socket_handler = this->socket_handlers.at(this->fds[i].fd);
socket_handler->on_send();
res--;
}
}
#endif
return true;
}
|
