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
|
/*
* Copyright (C) 2003-2013 The Music Player Daemon Project
* http://www.musicpd.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#ifndef MPD_MULTI_SOCKET_MONITOR_HXX
#define MPD_MULTI_SOCKET_MONITOR_HXX
#include "check.h"
#include "gcc.h"
#include "glib_compat.h"
#include <glib.h>
#include <forward_list>
#include <assert.h>
#include <stdint.h>
#ifdef WIN32
/* ERRORis a WIN32 macro that poisons our namespace; this is a
kludge to allow us to use it anyway */
#ifdef ERROR
#undef ERROR
#endif
#endif
class EventLoop;
/**
* Monitor multiple sockets.
*/
class MultiSocketMonitor {
struct Source {
GSource base;
MultiSocketMonitor *monitor;
};
EventLoop &loop;
Source *source;
uint64_t absolute_timeout_us;
std::forward_list<GPollFD> fds;
public:
static constexpr unsigned READ = G_IO_IN;
static constexpr unsigned WRITE = G_IO_OUT;
static constexpr unsigned ERROR = G_IO_ERR;
static constexpr unsigned HANGUP = G_IO_HUP;
MultiSocketMonitor(EventLoop &_loop);
~MultiSocketMonitor();
EventLoop &GetEventLoop() {
return loop;
}
gcc_pure
uint64_t GetTime() const {
return g_source_get_time(&source->base);
}
void InvalidateSockets() {
/* no-op because GLib always calls the GSource's
"prepare" method before each poll() anyway */
}
void AddSocket(int fd, unsigned events) {
fds.push_front({fd, gushort(events), 0});
g_source_add_poll(&source->base, &fds.front());
}
template<typename E>
void UpdateSocketList(E &&e) {
for (auto prev = fds.before_begin(), end = fds.end(),
i = std::next(prev);
i != end; i = std::next(prev)) {
assert(i->events != 0);
unsigned events = e(i->fd);
if (events != 0) {
i->events = events;
prev = i;
} else {
g_source_remove_poll(&source->base, &*i);
fds.erase_after(prev);
}
}
}
protected:
/**
* @return timeout [ms] or -1 for no timeout
*/
virtual int PrepareSockets() = 0;
virtual void DispatchSockets() = 0;
public:
/* GSource callbacks */
static gboolean Prepare(GSource *source, gint *timeout_r);
static gboolean Check(GSource *source);
static gboolean Dispatch(GSource *source, GSourceFunc callback,
gpointer user_data);
private:
bool Prepare(gint *timeout_r);
bool Check() const;
void Dispatch() {
DispatchSockets();
}
};
#endif
|
