165 lines
4.4 KiB
C++
165 lines
4.4 KiB
C++
/*
|
|
* Copyright 2003-2021 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.
|
|
*/
|
|
|
|
#include "AvahiPoll.hxx"
|
|
#include "event/SocketMonitor.hxx"
|
|
#include "event/TimerEvent.hxx"
|
|
#include "time/Convert.hxx"
|
|
|
|
static unsigned
|
|
FromAvahiWatchEvent(AvahiWatchEvent e)
|
|
{
|
|
return (e & AVAHI_WATCH_IN ? SocketMonitor::READ : 0) |
|
|
(e & AVAHI_WATCH_OUT ? SocketMonitor::WRITE : 0);
|
|
}
|
|
|
|
static AvahiWatchEvent
|
|
ToAvahiWatchEvent(unsigned e)
|
|
{
|
|
return AvahiWatchEvent((e & SocketMonitor::READ ? AVAHI_WATCH_IN : 0) |
|
|
(e & SocketMonitor::WRITE ? AVAHI_WATCH_OUT : 0) |
|
|
(e & SocketMonitor::ERROR ? AVAHI_WATCH_ERR : 0) |
|
|
(e & SocketMonitor::HANGUP ? AVAHI_WATCH_HUP : 0));
|
|
}
|
|
|
|
struct AvahiWatch final : private SocketMonitor {
|
|
private:
|
|
const AvahiWatchCallback callback;
|
|
void *const userdata;
|
|
|
|
AvahiWatchEvent received;
|
|
|
|
public:
|
|
AvahiWatch(SocketDescriptor _fd, AvahiWatchEvent _event,
|
|
AvahiWatchCallback _callback, void *_userdata,
|
|
EventLoop &_loop)
|
|
:SocketMonitor(_fd, _loop),
|
|
callback(_callback), userdata(_userdata),
|
|
received(AvahiWatchEvent(0)) {
|
|
Schedule(FromAvahiWatchEvent(_event));
|
|
}
|
|
|
|
static void WatchUpdate(AvahiWatch *w, AvahiWatchEvent event) {
|
|
w->Schedule(FromAvahiWatchEvent(event));
|
|
}
|
|
|
|
static AvahiWatchEvent WatchGetEvents(AvahiWatch *w) {
|
|
return w->received;
|
|
}
|
|
|
|
static void WatchFree(AvahiWatch *w) {
|
|
delete w;
|
|
}
|
|
|
|
private:
|
|
/* virtual methods from class SocketMonitor */
|
|
bool OnSocketReady(unsigned flags) noexcept override {
|
|
received = ToAvahiWatchEvent(flags);
|
|
callback(this, GetSocket().Get(), received, userdata);
|
|
received = AvahiWatchEvent(0);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
struct AvahiTimeout final {
|
|
TimerEvent event;
|
|
|
|
const AvahiTimeoutCallback callback;
|
|
void *const userdata;
|
|
|
|
public:
|
|
AvahiTimeout(const struct timeval *tv,
|
|
AvahiTimeoutCallback _callback, void *_userdata,
|
|
EventLoop &_loop)
|
|
:event(_loop, BIND_THIS_METHOD(OnTimeout)),
|
|
callback(_callback), userdata(_userdata) {
|
|
if (tv != nullptr)
|
|
Schedule(*tv);
|
|
}
|
|
|
|
static void TimeoutUpdate(AvahiTimeout *t, const struct timeval *tv) {
|
|
if (tv != nullptr)
|
|
t->Schedule(*tv);
|
|
else
|
|
t->event.Cancel();
|
|
}
|
|
|
|
static void TimeoutFree(AvahiTimeout *t) {
|
|
delete t;
|
|
}
|
|
|
|
private:
|
|
[[gnu::pure]]
|
|
Event::Duration AbsoluteToDuration(const struct timeval &tv) noexcept {
|
|
if (tv.tv_sec == 0)
|
|
/* schedule immediately */
|
|
return {};
|
|
|
|
struct timeval now;
|
|
if (gettimeofday(&now, nullptr) < 0)
|
|
/* shouldn't ever fail, but if it does, do
|
|
something reasonable */
|
|
return std::chrono::seconds(1);
|
|
|
|
auto d = ToSteadyClockDuration(tv)
|
|
- ToSteadyClockDuration(now);
|
|
if (d.count() < 0)
|
|
return {};
|
|
|
|
return d;
|
|
}
|
|
|
|
void Schedule(const struct timeval &tv) noexcept {
|
|
event.Schedule(AbsoluteToDuration(tv));
|
|
}
|
|
|
|
void OnTimeout() noexcept {
|
|
callback(this, userdata);
|
|
}
|
|
};
|
|
|
|
MyAvahiPoll::MyAvahiPoll(EventLoop &_loop):event_loop(_loop)
|
|
{
|
|
watch_new = WatchNew;
|
|
watch_update = AvahiWatch::WatchUpdate;
|
|
watch_get_events = AvahiWatch::WatchGetEvents;
|
|
watch_free = AvahiWatch::WatchFree;
|
|
timeout_new = TimeoutNew;
|
|
timeout_update = AvahiTimeout::TimeoutUpdate;
|
|
timeout_free = AvahiTimeout::TimeoutFree;
|
|
}
|
|
|
|
AvahiWatch *
|
|
MyAvahiPoll::WatchNew(const AvahiPoll *api, int fd, AvahiWatchEvent event,
|
|
AvahiWatchCallback callback, void *userdata) {
|
|
const MyAvahiPoll &poll = *(const MyAvahiPoll *)api;
|
|
|
|
return new AvahiWatch(SocketDescriptor(fd), event, callback, userdata,
|
|
poll.event_loop);
|
|
}
|
|
|
|
AvahiTimeout *
|
|
MyAvahiPoll::TimeoutNew(const AvahiPoll *api, const struct timeval *tv,
|
|
AvahiTimeoutCallback callback, void *userdata) {
|
|
const MyAvahiPoll &poll = *(const MyAvahiPoll *)api;
|
|
|
|
return new AvahiTimeout(tv, callback, userdata,
|
|
poll.event_loop);
|
|
}
|