kwin/plugins/idletime/poller.cpp
Martin Gräßlin ed3f55b953 [wayland] Fix heap-use-after-free in idle time plugin
We need to also destroy all idle objects when we tear down the
internal client connection. Otherwise the cleanup tries to free
the object after our Wayland connection is already destroyed.
2016-02-02 12:00:12 +01:00

134 lines
3.5 KiB
C++

/********************************************************************
KWin - the KDE window manager
This file is part of the KDE project.
Copyright (C) 2015 Martin Gräßlin <mgraesslin@kde.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, see <http://www.gnu.org/licenses/>.
*********************************************************************/
#include "poller.h"
#include "../../wayland_server.h"
#include <KWayland/Client/idle.h>
#include <KWayland/Client/registry.h>
#include <KWayland/Client/seat.h>
Poller::Poller(QObject *parent)
: AbstractSystemPoller(parent)
{
connect(KWin::waylandServer(), &KWin::WaylandServer::terminatingInternalClientConnection, this,
[this] {
qDeleteAll(m_timeouts);
m_timeouts.clear();
delete m_seat;
m_seat = nullptr;
delete m_idle;
m_idle = nullptr;
}
);
}
Poller::~Poller() = default;
bool Poller::isAvailable()
{
return true;
}
bool Poller::setUpPoller()
{
auto registry = KWin::waylandServer()->internalClientRegistry();
if (!m_seat) {
const auto iface = registry->interface(KWayland::Client::Registry::Interface::Seat);
m_seat = registry->createSeat(iface.name, iface.version, this);
}
if (!m_idle) {
const auto iface = registry->interface(KWayland::Client::Registry::Interface::Idle);
m_idle = registry->createIdle(iface.name, iface.version, this);
}
return m_seat->isValid() && m_idle->isValid();
}
void Poller::unloadPoller()
{
}
void Poller::addTimeout(int nextTimeout)
{
if (m_timeouts.contains(nextTimeout)) {
return;
}
if (!m_idle) {
return;
}
auto timeout = m_idle->getTimeout(nextTimeout, m_seat, this);
m_timeouts.insert(nextTimeout, timeout);
connect(timeout, &KWayland::Client::IdleTimeout::idle, this,
[this, nextTimeout] {
emit timeoutReached(nextTimeout);
}
);
connect(timeout, &KWayland::Client::IdleTimeout::resumeFromIdle, this, &Poller::resumingFromIdle);
}
void Poller::removeTimeout(int nextTimeout)
{
auto it = m_timeouts.find(nextTimeout);
if (it == m_timeouts.end()) {
return;
}
delete it.value();
m_timeouts.erase(it);
}
QList< int > Poller::timeouts() const
{
return QList<int>();
}
void Poller::catchIdleEvent()
{
if (m_catchResumeTimeout) {
// already setup
return;
}
if (!m_idle) {
return;
}
m_catchResumeTimeout = m_idle->getTimeout(0, m_seat, this);
connect(m_catchResumeTimeout, &KWayland::Client::IdleTimeout::resumeFromIdle, this,
[this] {
stopCatchingIdleEvents();
emit resumingFromIdle();
}
);
}
void Poller::stopCatchingIdleEvents()
{
delete m_catchResumeTimeout;
m_catchResumeTimeout = nullptr;
}
int Poller::forcePollRequest()
{
return 0;
}
void Poller::simulateUserActivity()
{
for (auto it = m_timeouts.constBegin(); it != m_timeouts.constEnd(); ++it) {
it.value()->simulateUserActivity();
}
}