event/SignalMonitor: use signalfd() if available
This commit is contained in:
@@ -288,6 +288,7 @@ libsystem_a_SOURCES = \
|
|||||||
src/system/SocketUtil.cxx src/system/SocketUtil.hxx \
|
src/system/SocketUtil.cxx src/system/SocketUtil.hxx \
|
||||||
src/system/SocketError.hxx \
|
src/system/SocketError.hxx \
|
||||||
src/system/resolver.c src/system/resolver.h \
|
src/system/resolver.c src/system/resolver.h \
|
||||||
|
src/system/SignalFD.cxx src/system/SignalFD.hxx \
|
||||||
src/system/clock.c src/system/clock.h
|
src/system/clock.c src/system/clock.h
|
||||||
|
|
||||||
# Event loop library
|
# Event loop library
|
||||||
|
@@ -141,6 +141,7 @@ AC_SEARCH_LIBS([gethostbyname], [nsl])
|
|||||||
|
|
||||||
AC_CHECK_FUNCS(pipe2 accept4)
|
AC_CHECK_FUNCS(pipe2 accept4)
|
||||||
MPD_OPTIONAL_FUNC(eventfd, USE_EVENTFD)
|
MPD_OPTIONAL_FUNC(eventfd, USE_EVENTFD)
|
||||||
|
MPD_OPTIONAL_FUNC(signalfd, USE_SIGNALFD)
|
||||||
|
|
||||||
AC_SEARCH_LIBS([exp], [m],,
|
AC_SEARCH_LIBS([exp], [m],,
|
||||||
[AC_MSG_ERROR([exp() not found])])
|
[AC_MSG_ERROR([exp() not found])])
|
||||||
|
@@ -27,27 +27,58 @@
|
|||||||
#include "util/Manual.hxx"
|
#include "util/Manual.hxx"
|
||||||
#include "system/FatalError.hxx"
|
#include "system/FatalError.hxx"
|
||||||
|
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
#include "system/SignalFD.hxx"
|
||||||
|
#else
|
||||||
|
#include "WakeFD.hxx"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifndef USE_SIGNALFD
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
#endif
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
|
|
||||||
class SignalMonitor final : private SocketMonitor {
|
class SignalMonitor final : private SocketMonitor {
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
SignalFD fd;
|
||||||
|
#else
|
||||||
WakeFD fd;
|
WakeFD fd;
|
||||||
|
#endif
|
||||||
|
|
||||||
public:
|
public:
|
||||||
SignalMonitor(EventLoop &_loop)
|
SignalMonitor(EventLoop &_loop)
|
||||||
:SocketMonitor(_loop) {
|
:SocketMonitor(_loop) {
|
||||||
|
#ifndef USE_SIGNALFD
|
||||||
SocketMonitor::Open(fd.Get());
|
SocketMonitor::Open(fd.Get());
|
||||||
SocketMonitor::ScheduleRead();
|
SocketMonitor::ScheduleRead();
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
~SignalMonitor() {
|
~SignalMonitor() {
|
||||||
/* prevent the descriptor to be closed twice */
|
/* prevent the descriptor to be closed twice */
|
||||||
SocketMonitor::Steal();
|
#ifdef USE_SIGNALFD
|
||||||
|
if (SocketMonitor::IsDefined())
|
||||||
|
#endif
|
||||||
|
SocketMonitor::Steal();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
void Update(sigset_t &mask) {
|
||||||
|
const bool was_open = SocketMonitor::IsDefined();
|
||||||
|
|
||||||
|
fd.Create(mask);
|
||||||
|
|
||||||
|
if (!was_open) {
|
||||||
|
SocketMonitor::Open(fd.Get());
|
||||||
|
SocketMonitor::ScheduleRead();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#else
|
||||||
void WakeUp() {
|
void WakeUp() {
|
||||||
fd.Write();
|
fd.Write();
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
private:
|
private:
|
||||||
virtual bool OnSocketReady(unsigned flags) override;
|
virtual bool OnSocketReady(unsigned flags) override;
|
||||||
@@ -57,10 +88,16 @@ private:
|
|||||||
static constexpr unsigned MAX_SIGNAL = 64;
|
static constexpr unsigned MAX_SIGNAL = 64;
|
||||||
|
|
||||||
static SignalHandler signal_handlers[MAX_SIGNAL];
|
static SignalHandler signal_handlers[MAX_SIGNAL];
|
||||||
|
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
static sigset_t signal_mask;
|
||||||
|
#else
|
||||||
static std::atomic_bool signal_pending[MAX_SIGNAL];
|
static std::atomic_bool signal_pending[MAX_SIGNAL];
|
||||||
|
#endif
|
||||||
|
|
||||||
static Manual<SignalMonitor> monitor;
|
static Manual<SignalMonitor> monitor;
|
||||||
|
|
||||||
|
#ifndef USE_SIGNALFD
|
||||||
static void
|
static void
|
||||||
SignalCallback(int signo)
|
SignalCallback(int signo)
|
||||||
{
|
{
|
||||||
@@ -69,13 +106,20 @@ SignalCallback(int signo)
|
|||||||
if (!signal_pending[signo].exchange(true))
|
if (!signal_pending[signo].exchange(true))
|
||||||
monitor->WakeUp();
|
monitor->WakeUp();
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
void
|
void
|
||||||
SignalMonitorInit(EventLoop &loop)
|
SignalMonitorInit(EventLoop &loop)
|
||||||
{
|
{
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
sigemptyset(&signal_mask);
|
||||||
|
#endif
|
||||||
|
|
||||||
monitor.Construct(loop);
|
monitor.Construct(loop);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#ifndef USE_SIGNALFD
|
||||||
|
|
||||||
static void
|
static void
|
||||||
x_sigaction(int signum, const struct sigaction &act)
|
x_sigaction(int signum, const struct sigaction &act)
|
||||||
{
|
{
|
||||||
@@ -83,9 +127,14 @@ x_sigaction(int signum, const struct sigaction &act)
|
|||||||
FatalSystemError("sigaction() failed");
|
FatalSystemError("sigaction() failed");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
void
|
void
|
||||||
SignalMonitorFinish()
|
SignalMonitorFinish()
|
||||||
{
|
{
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
std::fill_n(signal_handlers, MAX_SIGNAL, nullptr);
|
||||||
|
#else
|
||||||
struct sigaction sa;
|
struct sigaction sa;
|
||||||
sa.sa_flags = 0;
|
sa.sa_flags = 0;
|
||||||
sigemptyset(&sa.sa_mask);
|
sigemptyset(&sa.sa_mask);
|
||||||
@@ -99,6 +148,7 @@ SignalMonitorFinish()
|
|||||||
}
|
}
|
||||||
|
|
||||||
std::fill_n(signal_pending, MAX_SIGNAL, false);
|
std::fill_n(signal_pending, MAX_SIGNAL, false);
|
||||||
|
#endif
|
||||||
|
|
||||||
monitor.Destruct();
|
monitor.Destruct();
|
||||||
}
|
}
|
||||||
@@ -107,25 +157,46 @@ void
|
|||||||
SignalMonitorRegister(int signo, SignalHandler handler)
|
SignalMonitorRegister(int signo, SignalHandler handler)
|
||||||
{
|
{
|
||||||
assert(signal_handlers[signo] == nullptr);
|
assert(signal_handlers[signo] == nullptr);
|
||||||
|
#ifndef USE_SIGNALFD
|
||||||
assert(!signal_pending[signo]);
|
assert(!signal_pending[signo]);
|
||||||
|
#endif
|
||||||
|
|
||||||
signal_handlers[signo] = handler;
|
signal_handlers[signo] = handler;
|
||||||
|
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
sigaddset(&signal_mask, signo);
|
||||||
|
|
||||||
|
if (sigprocmask(SIG_BLOCK, &signal_mask, nullptr) < 0)
|
||||||
|
FatalSystemError("sigprocmask() failed");
|
||||||
|
|
||||||
|
monitor->Update(signal_mask);
|
||||||
|
#else
|
||||||
struct sigaction sa;
|
struct sigaction sa;
|
||||||
sa.sa_flags = 0;
|
sa.sa_flags = 0;
|
||||||
sigemptyset(&sa.sa_mask);
|
sigemptyset(&sa.sa_mask);
|
||||||
sa.sa_handler = SignalCallback;
|
sa.sa_handler = SignalCallback;
|
||||||
x_sigaction(signo, sa);
|
x_sigaction(signo, sa);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
bool
|
bool
|
||||||
SignalMonitor::OnSocketReady(unsigned)
|
SignalMonitor::OnSocketReady(unsigned)
|
||||||
{
|
{
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
int signo;
|
||||||
|
while ((signo = fd.Read()) >= 0) {
|
||||||
|
assert(unsigned(signo) < MAX_SIGNAL);
|
||||||
|
assert(signal_handlers[signo] != nullptr);
|
||||||
|
|
||||||
|
signal_handlers[signo]();
|
||||||
|
}
|
||||||
|
#else
|
||||||
fd.Read();
|
fd.Read();
|
||||||
|
|
||||||
for (unsigned i = 0; i < MAX_SIGNAL; ++i)
|
for (unsigned i = 0; i < MAX_SIGNAL; ++i)
|
||||||
if (signal_pending[i].exchange(false))
|
if (signal_pending[i].exchange(false))
|
||||||
signal_handlers[i]();
|
signal_handlers[i]();
|
||||||
|
#endif
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
58
src/system/SignalFD.cxx
Normal file
58
src/system/SignalFD.cxx
Normal file
@@ -0,0 +1,58 @@
|
|||||||
|
/*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "config.h"
|
||||||
|
#ifdef USE_SIGNALFD
|
||||||
|
#include "SignalFD.hxx"
|
||||||
|
#include "fd_util.h"
|
||||||
|
#include "FatalError.hxx"
|
||||||
|
|
||||||
|
#include <assert.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <sys/signalfd.h>
|
||||||
|
|
||||||
|
void
|
||||||
|
SignalFD::Create(const sigset_t &mask)
|
||||||
|
{
|
||||||
|
fd = ::signalfd(fd, &mask, SFD_NONBLOCK|SFD_CLOEXEC);
|
||||||
|
if (fd < 0)
|
||||||
|
FatalSystemError("signalfd() failed");
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
SignalFD::Close()
|
||||||
|
{
|
||||||
|
if (fd >= 0) {
|
||||||
|
::close(fd);
|
||||||
|
fd = -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
SignalFD::Read()
|
||||||
|
{
|
||||||
|
assert(fd >= 0);
|
||||||
|
|
||||||
|
signalfd_siginfo info;
|
||||||
|
return read(fd, &info, sizeof(info)) > 0
|
||||||
|
? info.ssi_signo
|
||||||
|
: -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif /* USE_SIGNALFD */
|
62
src/system/SignalFD.hxx
Normal file
62
src/system/SignalFD.hxx
Normal file
@@ -0,0 +1,62 @@
|
|||||||
|
/*
|
||||||
|
* 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_SIGNAL_FD_HXX
|
||||||
|
#define MPD_SIGNAL_FD_HXX
|
||||||
|
|
||||||
|
#include "check.h"
|
||||||
|
|
||||||
|
#include <signal.h>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A class that wraps signalfd().
|
||||||
|
*/
|
||||||
|
class SignalFD {
|
||||||
|
int fd;
|
||||||
|
|
||||||
|
public:
|
||||||
|
SignalFD():fd(-1) {}
|
||||||
|
~SignalFD() {
|
||||||
|
Close();
|
||||||
|
}
|
||||||
|
|
||||||
|
SignalFD(const SignalFD &other) = delete;
|
||||||
|
SignalFD &operator=(const SignalFD &other) = delete;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create the signalfd or update its mask.
|
||||||
|
*
|
||||||
|
* All errors are fatal.
|
||||||
|
*/
|
||||||
|
void Create(const sigset_t &mask);
|
||||||
|
void Close();
|
||||||
|
|
||||||
|
int Get() const {
|
||||||
|
return fd;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read the next signal from the file descriptor. Returns the
|
||||||
|
* signal number on success or -1 if there are no more
|
||||||
|
* signals.
|
||||||
|
*/
|
||||||
|
int Read();
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
Reference in New Issue
Block a user