2013-01-15 07:14:36 +01:00
|
|
|
/*
|
|
|
|
* 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"
|
2013-08-07 22:16:59 +02:00
|
|
|
#include "IdleMonitor.hxx"
|
|
|
|
#include "TimeoutMonitor.hxx"
|
|
|
|
#include "SocketMonitor.hxx"
|
2014-01-05 00:39:24 +01:00
|
|
|
#include "Compiler.h"
|
2013-08-07 22:16:59 +02:00
|
|
|
|
2013-01-15 07:14:36 +01:00
|
|
|
#include <forward_list>
|
2013-11-28 11:50:54 +01:00
|
|
|
#include <iterator>
|
2013-01-15 07:14:36 +01:00
|
|
|
|
|
|
|
#include <assert.h>
|
|
|
|
|
|
|
|
#ifdef WIN32
|
2013-11-24 16:07:12 +01:00
|
|
|
/* ERROR is a WIN32 macro that poisons our namespace; this is a kludge
|
|
|
|
to allow us to use it anyway */
|
2013-01-15 07:14:36 +01:00
|
|
|
#ifdef ERROR
|
|
|
|
#undef ERROR
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
2014-01-05 01:28:36 +01:00
|
|
|
#ifndef WIN32
|
|
|
|
struct pollfd;
|
|
|
|
#endif
|
|
|
|
|
2013-01-15 07:14:36 +01:00
|
|
|
class EventLoop;
|
|
|
|
|
|
|
|
/**
|
2013-11-24 19:28:04 +01:00
|
|
|
* Similar to #SocketMonitor, but monitors multiple sockets. To use
|
|
|
|
* it, implement the methods PrepareSockets() and DispatchSockets().
|
|
|
|
* In PrepareSockets(), use UpdateSocketList() and AddSocket().
|
|
|
|
* DispatchSockets() will be called if at least one socket is ready.
|
2013-01-15 07:14:36 +01:00
|
|
|
*/
|
2014-01-04 19:31:23 +01:00
|
|
|
class MultiSocketMonitor : IdleMonitor, TimeoutMonitor
|
2013-08-07 22:16:59 +02:00
|
|
|
{
|
|
|
|
class SingleFD final : public SocketMonitor {
|
|
|
|
MultiSocketMonitor &multi;
|
|
|
|
|
|
|
|
unsigned revents;
|
|
|
|
|
|
|
|
public:
|
|
|
|
SingleFD(MultiSocketMonitor &_multi, int _fd, unsigned events)
|
|
|
|
:SocketMonitor(_fd, _multi.GetEventLoop()),
|
|
|
|
multi(_multi), revents(0) {
|
|
|
|
Schedule(events);
|
|
|
|
}
|
|
|
|
|
|
|
|
int GetFD() const {
|
|
|
|
return SocketMonitor::Get();
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned GetEvents() const {
|
|
|
|
return SocketMonitor::GetScheduledFlags();
|
|
|
|
}
|
|
|
|
|
|
|
|
void SetEvents(unsigned _events) {
|
|
|
|
revents &= _events;
|
|
|
|
SocketMonitor::Schedule(_events);
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned GetReturnedEvents() const {
|
|
|
|
return revents;
|
|
|
|
}
|
|
|
|
|
|
|
|
void ClearReturnedEvents() {
|
|
|
|
revents = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected:
|
|
|
|
virtual bool OnSocketReady(unsigned flags) override {
|
|
|
|
revents = flags;
|
|
|
|
multi.SetReady();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
friend class SingleFD;
|
|
|
|
|
|
|
|
bool ready, refresh;
|
|
|
|
|
2013-08-10 10:10:40 +02:00
|
|
|
std::forward_list<SingleFD> fds;
|
2013-01-15 07:14:36 +01:00
|
|
|
|
|
|
|
public:
|
2013-08-07 22:16:59 +02:00
|
|
|
static constexpr unsigned READ = SocketMonitor::READ;
|
|
|
|
static constexpr unsigned WRITE = SocketMonitor::WRITE;
|
|
|
|
static constexpr unsigned ERROR = SocketMonitor::ERROR;
|
|
|
|
static constexpr unsigned HANGUP = SocketMonitor::HANGUP;
|
2013-01-15 07:14:36 +01:00
|
|
|
|
|
|
|
MultiSocketMonitor(EventLoop &_loop);
|
|
|
|
~MultiSocketMonitor();
|
|
|
|
|
2013-08-07 22:16:59 +02:00
|
|
|
using IdleMonitor::GetEventLoop;
|
2013-08-08 21:49:17 +02:00
|
|
|
|
2013-08-07 22:16:59 +02:00
|
|
|
public:
|
2013-01-15 07:14:36 +01:00
|
|
|
void InvalidateSockets() {
|
2013-08-07 22:16:59 +02:00
|
|
|
refresh = true;
|
|
|
|
IdleMonitor::Schedule();
|
2013-01-15 07:14:36 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
void AddSocket(int fd, unsigned events) {
|
2013-08-10 10:10:40 +02:00
|
|
|
fds.emplace_front(*this, fd, events);
|
2013-01-15 07:14:36 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
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)) {
|
2013-08-10 10:10:40 +02:00
|
|
|
assert(i->GetEvents() != 0);
|
2013-01-15 07:14:36 +01:00
|
|
|
|
2013-08-10 10:10:40 +02:00
|
|
|
unsigned events = e(i->GetFD());
|
2013-01-15 07:14:36 +01:00
|
|
|
if (events != 0) {
|
2013-08-10 10:10:40 +02:00
|
|
|
i->SetEvents(events);
|
2013-01-15 07:14:36 +01:00
|
|
|
prev = i;
|
|
|
|
} else {
|
2013-08-07 22:16:59 +02:00
|
|
|
i->Steal();
|
2013-01-15 07:14:36 +01:00
|
|
|
fds.erase_after(prev);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-05 01:28:36 +01:00
|
|
|
#ifndef WIN32
|
|
|
|
/**
|
|
|
|
* Replace the socket list with the given file descriptors.
|
|
|
|
* The given pollfd array will be modified by this method.
|
|
|
|
*/
|
|
|
|
void ReplaceSocketList(pollfd *pfds, unsigned n);
|
|
|
|
#endif
|
|
|
|
|
2013-01-15 07:14:36 +01:00
|
|
|
protected:
|
2013-08-10 10:57:00 +02:00
|
|
|
/**
|
|
|
|
* @return timeout [ms] or -1 for no timeout
|
|
|
|
*/
|
|
|
|
virtual int PrepareSockets() = 0;
|
2013-01-15 07:14:36 +01:00
|
|
|
virtual void DispatchSockets() = 0;
|
|
|
|
|
2013-08-07 22:16:59 +02:00
|
|
|
private:
|
|
|
|
void SetReady() {
|
|
|
|
ready = true;
|
|
|
|
IdleMonitor::Schedule();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Prepare();
|
|
|
|
|
|
|
|
virtual void OnTimeout() final {
|
|
|
|
SetReady();
|
|
|
|
IdleMonitor::Schedule();
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void OnIdle() final;
|
2013-01-15 07:14:36 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
#endif
|