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
|
/*
* 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_SOCKET_MONITOR_HXX
#define MPD_SOCKET_MONITOR_HXX
#include "check.h"
#include <glib.h>
#include <assert.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;
class SocketMonitor {
struct Source {
GSource base;
SocketMonitor *monitor;
};
int fd;
EventLoop &loop;
Source *source;
GPollFD poll;
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;
SocketMonitor(EventLoop &_loop)
:fd(-1), loop(_loop), source(nullptr) {}
SocketMonitor(int _fd, EventLoop &_loop);
~SocketMonitor();
bool IsDefined() const {
return fd >= 0;
}
int Get() const {
assert(IsDefined());
return fd;
}
void Open(int _fd);
/**
* "Steal" the socket descriptor. This abandons the socket
* and puts the responsibility for closing it to the caller.
*/
int Steal();
void Close();
void Schedule(unsigned flags) {
poll.events = flags;
poll.revents &= flags;
}
void Cancel() {
poll.events = 0;
}
void ScheduleRead() {
poll.events |= READ|HANGUP|ERROR;
}
void ScheduleWrite() {
poll.events |= WRITE;
}
void CancelRead() {
poll.events &= ~(READ|HANGUP|ERROR);
}
void CancelWrite() {
poll.events &= ~WRITE;
}
protected:
virtual void OnSocketReady(unsigned flags) = 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 Check() const {
return (poll.revents & poll.events) != 0;
}
void Dispatch() {
OnSocketReady(poll.revents & poll.events);
}
};
#endif
|
