wayland_server.cpp 24.9 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
/********************************************************************
 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 "wayland_server.h"
21
#include "client.h"
22
#include "platform.h"
23
#include "composite.h"
24
#include "idle_inhibition.h"
25
#include "screens.h"
26
#include "shell_client.h"
27
#include "workspace.h"
28

29 30
// Client
#include <KWayland/Client/connection_thread.h>
31
#include <KWayland/Client/event_queue.h>
32
#include <KWayland/Client/registry.h>
33
#include <KWayland/Client/shm_pool.h>
34
#include <KWayland/Client/surface.h>
35
// Server
David Edmundson's avatar
David Edmundson committed
36
#include <KWayland/Server/appmenu_interface.h>
37
#include <KWayland/Server/compositor_interface.h>
38
#include <KWayland/Server/datadevicemanager_interface.h>
Martin Flöser's avatar
Martin Flöser committed
39
#include <KWayland/Server/datasource_interface.h>
40
#include <KWayland/Server/display.h>
41
#include <KWayland/Server/dpms_interface.h>
42
#include <KWayland/Server/idle_interface.h>
43
#include <KWayland/Server/idleinhibit_interface.h>
44
#include <KWayland/Server/output_interface.h>
45
#include <KWayland/Server/plasmashell_interface.h>
46
#include <KWayland/Server/plasmawindowmanagement_interface.h>
47
#include <KWayland/Server/pointerconstraints_interface.h>
48
#include <KWayland/Server/pointergestures_interface.h>
49
#include <KWayland/Server/qtsurfaceextension_interface.h>
50
#include <KWayland/Server/seat_interface.h>
51
#include <KWayland/Server/server_decoration_interface.h>
52
#include <KWayland/Server/shadow_interface.h>
53
#include <KWayland/Server/subcompositor_interface.h>
54
#include <KWayland/Server/blur_interface.h>
55
#include <KWayland/Server/shell_interface.h>
56 57
#include <KWayland/Server/outputmanagement_interface.h>
#include <KWayland/Server/outputconfiguration_interface.h>
58
#include <KWayland/Server/xdgshell_interface.h>
Marco Martin's avatar
Marco Martin committed
59
#include <KWayland/Server/xdgforeign_interface.h>
60

61
// Qt
62
#include <QThread>
63 64
#include <QWindow>

65 66 67
// system
#include <sys/types.h>
#include <sys/socket.h>
68
#include <unistd.h>
69

70 71 72
//screenlocker
#include <KScreenLocker/KsldApp>

73 74 75 76 77 78 79 80 81 82
using namespace KWayland::Server;

namespace KWin
{

KWIN_SINGLETON_FACTORY(WaylandServer)

WaylandServer::WaylandServer(QObject *parent)
    : QObject(parent)
{
83
    qRegisterMetaType<KWayland::Server::OutputInterface::DpmsMode>();
84 85

    connect(kwinApp(), &Application::screensCreated, this, &WaylandServer::initOutputs);
86
    connect(kwinApp(), &Application::x11ConnectionChanged, this, &WaylandServer::setupX11ClipboardSync);
87 88
}

89 90 91 92
WaylandServer::~WaylandServer()
{
    destroyInputMethodConnection();
}
93 94

void WaylandServer::destroyInternalConnection()
95
{
96
    emit terminatingInternalClientConnection();
97
    if (m_internalConnection.client) {
98 99 100 101 102 103 104 105 106
        // delete all connections hold by plugins like e.g. widget style
        const auto connections = KWayland::Client::ConnectionThread::connections();
        for (auto c : connections) {
            if (c == m_internalConnection.client) {
                continue;
            }
            emit c->connectionDied();
        }

107 108
        delete m_internalConnection.registry;
        delete m_internalConnection.shm;
109
        dispatch();
110 111 112
        m_internalConnection.client->deleteLater();
        m_internalConnection.clientThread->quit();
        m_internalConnection.clientThread->wait();
113
        delete m_internalConnection.clientThread;
114
        m_internalConnection.client = nullptr;
115
        m_internalConnection.server->destroy();
116
        m_internalConnection.server = nullptr;
117 118
    }
}
119

120 121 122 123
void WaylandServer::terminateClientConnections()
{
    destroyInternalConnection();
    destroyInputMethodConnection();
124 125 126 127 128
    if (m_display) {
        const auto connections = m_display->connections();
        for (auto it = connections.begin(); it != connections.end(); ++it) {
            (*it)->destroy();
        }
129 130 131
    }
}

132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
template <class T>
void WaylandServer::createSurface(T *surface)
{
    if (!Workspace::self()) {
        // it's possible that a Surface gets created before Workspace is created
        return;
    }
    if (surface->client() == m_xwayland.client) {
        // skip Xwayland clients, those are created using standard X11 way
        return;
    }
    if (surface->client() == m_screenLockerClientConnection) {
        ScreenLocker::KSldApp::self()->lockScreenShown();
    }
    auto client = new ShellClient(surface);
147 148 149 150 151 152 153 154 155
    auto it = std::find_if(m_plasmaShellSurfaces.begin(), m_plasmaShellSurfaces.end(),
        [client] (PlasmaShellSurfaceInterface *surface) {
            return client->surface() == surface->surface();
        }
    );
    if (it != m_plasmaShellSurfaces.end()) {
        client->installPlasmaShellSurface(*it);
        m_plasmaShellSurfaces.erase(it);
    }
David Edmundson's avatar
David Edmundson committed
156 157 158
    if (auto menu = m_appMenuManager->appMenuForSurface(surface->surface())) {
        client->installAppMenu(menu);
    }
159 160 161 162 163 164 165 166 167 168
    if (client->isInternal()) {
        m_internalClients << client;
    } else {
        m_clients << client;
    }
    if (client->readyForPainting()) {
        emit shellClientAdded(client);
    } else {
        connect(client, &ShellClient::windowShown, this, &WaylandServer::shellClientShown);
    }
Marco Martin's avatar
Marco Martin committed
169 170

    //not directly connected as the connection is tied to client instead of this
171
    connect(m_XdgForeign, &KWayland::Server::XdgForeignInterface::transientChanged, client, [this](KWayland::Server::SurfaceInterface *child) {
Marco Martin's avatar
Marco Martin committed
172 173
        emit foreignTransientChanged(child);
    });
174 175
}

176
bool WaylandServer::init(const QByteArray &socketName, InitalizationFlags flags)
177
{
178
    m_initFlags = flags;
179 180 181 182 183
    m_display = new KWayland::Server::Display(this);
    if (!socketName.isNull() && !socketName.isEmpty()) {
        m_display->setSocketName(QString::fromUtf8(socketName));
    }
    m_display->start();
184 185 186
    if (!m_display->isRunning()) {
        return false;
    }
187 188
    m_compositor = m_display->createCompositor(m_display);
    m_compositor->create();
189 190 191 192 193 194 195 196
    connect(m_compositor, &CompositorInterface::surfaceCreated, this,
        [this] (SurfaceInterface *surface) {
            // check whether we have a Toplevel with the Surface's id
            Workspace *ws = Workspace::self();
            if (!ws) {
                // it's possible that a Surface gets created before Workspace is created
                return;
            }
197 198 199 200
            if (surface->client() != xWaylandConnection()) {
                // setting surface is only relevat for Xwayland clients
                return;
            }
201 202 203 204 205 206 207 208
            auto check = [surface] (const Toplevel *t) {
                return t->surfaceId() == surface->id();
            };
            if (Toplevel *t = ws->findToplevel(check)) {
                t->setSurface(surface);
            }
        }
    );
209 210
    m_shell = m_display->createShell(m_display);
    m_shell->create();
211 212 213 214 215 216
    connect(m_shell, &ShellInterface::surfaceCreated, this, &WaylandServer::createSurface<ShellSurfaceInterface>);
    m_xdgShell = m_display->createXdgShell(XdgShellInterfaceVersion::UnstableV5, m_display);
    m_xdgShell->create();
    connect(m_xdgShell, &XdgShellInterface::surfaceCreated, this, &WaylandServer::createSurface<XdgShellSurfaceInterface>);
    // TODO: verify seat and serial
    connect(m_xdgShell, &XdgShellInterface::popupCreated, this, &WaylandServer::createSurface<XdgShellPopupInterface>);
David Edmundson's avatar
David Edmundson committed
217 218 219 220 221 222 223

    m_xdgShell6 = m_display->createXdgShell(XdgShellInterfaceVersion::UnstableV6, m_display);
    m_xdgShell6->create();
    connect(m_xdgShell6, &XdgShellInterface::surfaceCreated, this, &WaylandServer::createSurface<XdgShellSurfaceInterface>);
    connect(m_xdgShell6, &XdgShellInterface::xdgPopupCreated, this, &WaylandServer::createSurface<XdgShellPopupInterface>);


224
    m_display->createShm();
225 226
    m_seat = m_display->createSeat(m_display);
    m_seat->create();
227
    m_display->createPointerGestures(PointerGesturesInterfaceVersion::UnstableV1, m_display)->create();
228
    m_display->createPointerConstraints(PointerConstraintsInterfaceVersion::UnstableV1, m_display)->create();
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247
    auto ddm = m_display->createDataDeviceManager(m_display);
    ddm->create();
    connect(ddm, &DataDeviceManagerInterface::dataDeviceCreated, this,
        [this] (DataDeviceInterface *ddi) {
            if (ddi->client() == m_xclipbaordSync.client && m_xclipbaordSync.client != nullptr) {
                m_xclipbaordSync.ddi = QPointer<DataDeviceInterface>(ddi);
                connect(m_xclipbaordSync.ddi.data(), &DataDeviceInterface::selectionChanged, this,
                    [this] {
                        // testing whether the active client inherits Client
                        // it would be better to test for the keyboard focus, but we might get a clipboard update
                        // when the Client is already active, but no Surface is created yet.
                        if (workspace()->activeClient() && workspace()->activeClient()->inherits("KWin::Client")) {
                            m_seat->setSelection(m_xclipbaordSync.ddi.data());
                        }
                    }
                );
            }
        }
    );
248 249 250 251 252
    auto idle = m_display->createIdle(m_display);
    idle->create();
    auto idleInhibition = new IdleInhibition(idle);
    connect(this, &WaylandServer::shellClientAdded, idleInhibition, &IdleInhibition::registerShellClient);
    m_display->createIdleInhibitManager(IdleInhibitManagerInterfaceVersion::UnstableV1, m_display)->create();
253 254 255 256 257 258
    m_plasmaShell = m_display->createPlasmaShell(m_display);
    m_plasmaShell->create();
    connect(m_plasmaShell, &PlasmaShellInterface::surfaceCreated,
        [this] (PlasmaShellSurfaceInterface *surface) {
            if (ShellClient *client = findClient(surface->surface())) {
                client->installPlasmaShellSurface(surface);
259 260 261 262 263 264 265
            } else {
                m_plasmaShellSurfaces << surface;
                connect(surface, &QObject::destroyed, this,
                    [this, surface] {
                        m_plasmaShellSurfaces.removeOne(surface);
                    }
                );
266 267 268
            }
        }
    );
David Edmundson's avatar
David Edmundson committed
269 270


271 272 273 274 275 276 277 278 279
    m_qtExtendedSurface = m_display->createQtSurfaceExtension(m_display);
    m_qtExtendedSurface->create();
    connect(m_qtExtendedSurface, &QtSurfaceExtensionInterface::surfaceCreated,
        [this] (QtExtendedSurfaceInterface *surface) {
            if (ShellClient *client = findClient(surface->surface())) {
                client->installQtExtendedSurface(surface);
            }
        }
    );
David Edmundson's avatar
David Edmundson committed
280 281 282 283 284 285 286 287 288 289
    m_appMenuManager = m_display->createAppMenuManagerInterface(m_display);
    m_appMenuManager->create();
    connect(m_appMenuManager, &AppMenuManagerInterface::appMenuCreated,
        [this] (AppMenuInterface *appMenu) {
            if (ShellClient *client = findClient(appMenu->surface())) {
                client->installAppMenu(appMenu);
            }
        }
    );

290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315
    m_windowManagement = m_display->createPlasmaWindowManagement(m_display);
    m_windowManagement->create();
    m_windowManagement->setShowingDesktopState(PlasmaWindowManagementInterface::ShowingDesktopState::Disabled);
    connect(m_windowManagement, &PlasmaWindowManagementInterface::requestChangeShowingDesktop, this,
        [] (PlasmaWindowManagementInterface::ShowingDesktopState state) {
            if (!workspace()) {
                return;
            }
            bool set = false;
            switch (state) {
            case PlasmaWindowManagementInterface::ShowingDesktopState::Disabled:
                set = false;
                break;
            case PlasmaWindowManagementInterface::ShowingDesktopState::Enabled:
                set = true;
                break;
            default:
                Q_UNREACHABLE();
                break;
            }
            if (set == workspace()->showingDesktop()) {
                return;
            }
            workspace()->setShowingDesktop(set);
        }
    );
316 317
    auto shadowManager = m_display->createShadowManager(m_display);
    shadowManager->create();
318 319

    m_display->createDpmsManager(m_display)->create();
320 321 322 323 324 325 326 327 328 329

    m_decorationManager = m_display->createServerSideDecorationManager(m_display);
    connect(m_decorationManager, &ServerSideDecorationManagerInterface::decorationCreated, this,
        [this] (ServerSideDecorationInterface *deco) {
            if (ShellClient *c = findClient(deco->surface())) {
                c->installServerSideDecoration(deco);
            }
        }
    );
    m_decorationManager->create();
330 331 332 333

    m_outputManagement = m_display->createOutputManagement(m_display);
    connect(m_outputManagement, &OutputManagementInterface::configurationChangeRequested,
            this, [this](KWayland::Server::OutputConfigurationInterface *config) {
334
                kwinApp()->platform()->configurationChangeRequested(config);
335
    });
336
    m_outputManagement->create();
337 338

    m_display->createSubCompositor(m_display)->create();
339

Marco Martin's avatar
Marco Martin committed
340 341 342
    m_XdgForeign = m_display->createXdgForeignInterface(m_display);
    m_XdgForeign->create();

343
    return true;
344 345
}

Marco Martin's avatar
Marco Martin committed
346 347 348 349 350
SurfaceInterface *WaylandServer::findForeignTransientForSurface(SurfaceInterface *surface)
{
    return m_XdgForeign->transientFor(surface);
}

351 352 353 354 355 356 357 358 359 360 361
void WaylandServer::shellClientShown(Toplevel *t)
{
    ShellClient *c = dynamic_cast<ShellClient*>(t);
    if (!c) {
        qCWarning(KWIN_CORE) << "Failed to cast a Toplevel which is supposed to be a ShellClient to ShellClient";
        return;
    }
    disconnect(c, &ShellClient::windowShown, this, &WaylandServer::shellClientShown);
    emit shellClientAdded(c);
}

362 363 364 365 366 367 368 369 370 371 372 373 374
void WaylandServer::initWorkspace()
{
    if (m_windowManagement) {
        connect(workspace(), &Workspace::showingDesktopChanged, this,
            [this] (bool set) {
                using namespace KWayland::Server;
                m_windowManagement->setShowingDesktopState(set ?
                    PlasmaWindowManagementInterface::ShowingDesktopState::Enabled :
                    PlasmaWindowManagementInterface::ShowingDesktopState::Disabled
                );
            }
        );
    }
375

376
    if (hasScreenLockerIntegration()) {
377 378 379 380 381 382 383 384 385
        if (m_internalConnection.interfacesAnnounced) {
            initScreenLocker();
        } else {
            connect(m_internalConnection.registry, &KWayland::Client::Registry::interfacesAnnounced, this, &WaylandServer::initScreenLocker);
        }
    } else {
        emit initialized();
    }
}
386

387 388 389 390 391 392 393 394 395 396 397 398
void WaylandServer::initScreenLocker()
{
    ScreenLocker::KSldApp::self();
    ScreenLocker::KSldApp::self()->setWaylandDisplay(m_display);
    ScreenLocker::KSldApp::self()->setGreeterEnvironment(kwinApp()->processStartupEnvironment());
    ScreenLocker::KSldApp::self()->initialize();

    connect(ScreenLocker::KSldApp::self(), &ScreenLocker::KSldApp::greeterClientConnectionChanged, this,
        [this] () {
            m_screenLockerClientConnection = ScreenLocker::KSldApp::self()->greeterClientConnection();
        }
    );
399

400 401 402
    connect(ScreenLocker::KSldApp::self(), &ScreenLocker::KSldApp::unlocked, this,
        [this] () {
            m_screenLockerClientConnection = nullptr;
403
        }
404 405 406 407
    );

    if (m_initFlags.testFlag(InitalizationFlag::LockScreen)) {
        ScreenLocker::KSldApp::self()->lock(ScreenLocker::EstablishLock::Immediate);
408
    }
409
    emit initialized();
410
}
411

412 413
void WaylandServer::initOutputs()
{
414
    if (kwinApp()->platform()->handlesOutputs()) {
415 416
        return;
    }
417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432
    syncOutputsToWayland();
    connect(screens(), &Screens::changed, this,
        [this] {
            // when screens change we need to sync this to Wayland.
            // Unfortunately we don't have much information and cannot properly match a KWin screen
            // to a Wayland screen.
            // Thus we just recreate all outputs and delete the old ones
            const auto outputs = m_display->outputs();
            syncOutputsToWayland();
            qDeleteAll(outputs);
        }
    );
}

void WaylandServer::syncOutputsToWayland()
{
433 434 435 436
    Screens *s = screens();
    Q_ASSERT(s);
    for (int i = 0; i < s->count(); ++i) {
        OutputInterface *output = m_display->createOutput(m_display);
David Edmundson's avatar
David Edmundson committed
437
        output->setScale(s->scale(i));
438 439
        const QRect &geo = s->geometry(i);
        output->setGlobalPosition(geo.topLeft());
440
        output->setPhysicalSize(s->physicalSize(i).toSize());
441
        output->addMode(geo.size());
442 443
        output->create();
    }
444 445
}

446
WaylandServer::SocketPairConnection WaylandServer::createConnection()
447
{
448
    SocketPairConnection ret;
449 450 451
    int sx[2];
    if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, sx) < 0) {
        qCWarning(KWIN_CORE) << "Could not create socket";
452 453 454 455 456 457 458 459 460 461 462
        return ret;
    }
    ret.connection = m_display->createClient(sx[0]);
    ret.fd = sx[1];
    return ret;
}

int WaylandServer::createXWaylandConnection()
{
    const auto socket = createConnection();
    if (!socket.connection) {
463 464
        return -1;
    }
465
    m_xwayland.client = socket.connection;
466
    m_xwayland.destroyConnection = connect(m_xwayland.client, &KWayland::Server::ClientConnection::disconnected, this,
467 468 469 470
        [] {
            qFatal("Xwayland Connection died");
        }
    );
471
    return socket.fd;
472 473
}

474 475 476 477 478
void WaylandServer::destroyXWaylandConnection()
{
    if (!m_xwayland.client) {
        return;
    }
479 480 481 482
    // first terminate the clipboard sync
    if (m_xclipbaordSync.process) {
        m_xclipbaordSync.process->terminate();
    }
483 484 485 486 487
    disconnect(m_xwayland.destroyConnection);
    m_xwayland.client->destroy();
    m_xwayland.client = nullptr;
}

488 489
int WaylandServer::createInputMethodConnection()
{
490 491
    const auto socket = createConnection();
    if (!socket.connection) {
492 493
        return -1;
    }
494 495
    m_inputMethodServerConnection = socket.connection;
    return socket.fd;
496 497
}

498 499 500 501 502 503 504 505 506
void WaylandServer::destroyInputMethodConnection()
{
    if (!m_inputMethodServerConnection) {
        return;
    }
    m_inputMethodServerConnection->destroy();
    m_inputMethodServerConnection = nullptr;
}

507 508
int WaylandServer::createXclipboardSyncConnection()
{
509 510
    const auto socket = createConnection();
    if (!socket.connection) {
511 512
        return -1;
    }
513 514
    m_xclipbaordSync.client = socket.connection;
    return socket.fd;
515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551
}

void WaylandServer::setupX11ClipboardSync()
{
    if (m_xclipbaordSync.process) {
        return;
    }

    int socket = dup(createXclipboardSyncConnection());
    if (socket == -1) {
        delete m_xclipbaordSync.client;
        m_xclipbaordSync.client = nullptr;
        return;
    }
    if (socket >= 0) {
        QProcessEnvironment environment = kwinApp()->processStartupEnvironment();
        environment.insert(QStringLiteral("WAYLAND_SOCKET"), QByteArray::number(socket));
        environment.insert(QStringLiteral("DISPLAY"), QString::fromUtf8(qgetenv("DISPLAY")));
        environment.remove("WAYLAND_DISPLAY");
        m_xclipbaordSync.process = new Process(this);
        m_xclipbaordSync.process->setProcessChannelMode(QProcess::ForwardedErrorChannel);
        auto finishedSignal = static_cast<void (QProcess::*)(int, QProcess::ExitStatus)>(&QProcess::finished);
        connect(m_xclipbaordSync.process, finishedSignal, this,
            [this] {
                m_xclipbaordSync.process->deleteLater();
                m_xclipbaordSync.process = nullptr;
                m_xclipbaordSync.ddi.clear();
                m_xclipbaordSync.client->destroy();
                m_xclipbaordSync.client = nullptr;
                // TODO: restart
            }
        );
        m_xclipbaordSync.process->setProcessEnvironment(environment);
        m_xclipbaordSync.process->start(QStringLiteral(KWIN_XCLIPBOARD_SYNC_BIN));
    }
}

552 553
void WaylandServer::createInternalConnection()
{
554 555
    const auto socket = createConnection();
    if (!socket.connection) {
556 557
        return;
    }
558
    m_internalConnection.server = socket.connection;
559
    using namespace KWayland::Client;
560
    m_internalConnection.client = new ConnectionThread();
561
    m_internalConnection.client->setSocketFd(socket.fd);
562 563 564 565
    m_internalConnection.clientThread = new QThread;
    m_internalConnection.client->moveToThread(m_internalConnection.clientThread);
    m_internalConnection.clientThread->start();

566 567
    connect(m_internalConnection.client, &ConnectionThread::connected, this,
        [this] {
568 569 570 571
            Registry *registry = new Registry(this);
            EventQueue *eventQueue = new EventQueue(this);
            eventQueue->setup(m_internalConnection.client);
            registry->setEventQueue(eventQueue);
572
            registry->create(m_internalConnection.client);
573
            m_internalConnection.registry = registry;
574
            connect(registry, &Registry::shmAnnounced, this,
575
                [this] (quint32 name, quint32 version) {
576
                    m_internalConnection.shm = m_internalConnection.registry->createShmPool(name, version, this);
577 578
                }
            );
579 580 581 582 583
            connect(registry, &Registry::interfacesAnnounced, this,
                [this] {
                    m_internalConnection.interfacesAnnounced = true;
                }
            );
584 585 586 587 588 589
            registry->setup();
        }
    );
    m_internalConnection.client->initConnection();
}

590 591 592
void WaylandServer::removeClient(ShellClient *c)
{
    m_clients.removeAll(c);
593
    m_internalClients.removeAll(c);
594 595 596
    emit shellClientRemoved(c);
}

597 598 599 600 601
void WaylandServer::dispatch()
{
    if (!m_display) {
        return;
    }
602 603
    if (m_internalConnection.server) {
        m_internalConnection.server->flush();
604 605 606 607
    }
    m_display->dispatchEvents(0);
}

608
static ShellClient *findClientInList(const QList<ShellClient*> &clients, quint32 id)
609
{
610
    auto it = std::find_if(clients.begin(), clients.end(),
611 612 613 614
        [id] (ShellClient *c) {
            return c->windowId() == id;
        }
    );
615
    if (it == clients.end()) {
616 617 618 619 620
        return nullptr;
    }
    return *it;
}

621 622 623 624 625 626 627 628 629 630 631 632 633
static ShellClient *findClientInList(const QList<ShellClient*> &clients, KWayland::Server::SurfaceInterface *surface)
{
    auto it = std::find_if(clients.begin(), clients.end(),
        [surface] (ShellClient *c) {
            return c->surface() == surface;
        }
    );
    if (it == clients.end()) {
        return nullptr;
    }
    return *it;
}

634 635 636 637 638 639 640 641 642 643 644 645 646 647
ShellClient *WaylandServer::findClient(quint32 id) const
{
    if (id == 0) {
        return nullptr;
    }
    if (ShellClient *c = findClientInList(m_clients, id)) {
        return c;
    }
    if (ShellClient *c = findClientInList(m_internalClients, id)) {
        return c;
    }
    return nullptr;
}

648 649 650 651 652 653 654 655 656 657 658 659 660 661
ShellClient *WaylandServer::findClient(SurfaceInterface *surface) const
{
    if (!surface) {
        return nullptr;
    }
    if (ShellClient *c = findClientInList(m_clients, surface)) {
        return c;
    }
    if (ShellClient *c = findClientInList(m_internalClients, surface)) {
        return c;
    }
    return nullptr;
}

662 663 664 665 666
AbstractClient *WaylandServer::findAbstractClient(SurfaceInterface *surface) const
{
    return findClient(surface);
}

667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682
ShellClient *WaylandServer::findClient(QWindow *w) const
{
    if (!w) {
        return nullptr;
    }
    auto it = std::find_if(m_internalClients.constBegin(), m_internalClients.constEnd(),
        [w] (const ShellClient *c) {
            return c->internalWindow() == w;
        }
    );
    if (it != m_internalClients.constEnd()) {
        return *it;
    }
    return nullptr;
}

683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724
quint32 WaylandServer::createWindowId(SurfaceInterface *surface)
{
    auto it = m_clientIds.constFind(surface->client());
    quint16 clientId = 0;
    if (it != m_clientIds.constEnd()) {
        clientId = it.value();
    } else {
        clientId = createClientId(surface->client());
    }
    Q_ASSERT(clientId != 0);
    quint32 id = clientId;
    // TODO: this does not prevent that two surfaces of same client get same id
    id = (id << 16) | (surface->id() & 0xFFFF);
    if (findClient(id)) {
        qCWarning(KWIN_CORE) << "Invalid client windowId generated:" << id;
        return 0;
    }
    return id;
}

quint16 WaylandServer::createClientId(ClientConnection *c)
{
    auto ids = m_clientIds.values().toSet();
    quint16 id = 1;
    if (!ids.isEmpty()) {
        for (quint16 i = ids.count() + 1; i >= 1 ; i--) {
            if (!ids.contains(i)) {
                id = i;
                break;
            }
        }
    }
    Q_ASSERT(!ids.contains(id));
    m_clientIds.insert(c, id);
    connect(c, &ClientConnection::disconnected, this,
        [this] (ClientConnection *c) {
            m_clientIds.remove(c);
        }
    );
    return id;
}

725 726
bool WaylandServer::isScreenLocked() const
{
727 728 729
    if (!hasScreenLockerIntegration()) {
        return false;
    }
730 731
    return ScreenLocker::KSldApp::self()->lockState() == ScreenLocker::KSldApp::Locked ||
           ScreenLocker::KSldApp::self()->lockState() == ScreenLocker::KSldApp::AcquiringLock;
732 733
}

734 735 736 737 738
bool WaylandServer::hasScreenLockerIntegration() const
{
    return !m_initFlags.testFlag(InitalizationFlag::NoLockScreenIntegration);
}

739
}