wayland_server.cpp 25.5 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/server_decoration_palette_interface.h>
53
#include <KWayland/Server/shadow_interface.h>
54
#include <KWayland/Server/subcompositor_interface.h>
55
#include <KWayland/Server/blur_interface.h>
56
#include <KWayland/Server/shell_interface.h>
57 58
#include <KWayland/Server/outputmanagement_interface.h>
#include <KWayland/Server/outputconfiguration_interface.h>
59
#include <KWayland/Server/xdgshell_interface.h>
Marco Martin's avatar
Marco Martin committed
60
#include <KWayland/Server/xdgforeign_interface.h>
61

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

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

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

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

namespace KWin
{

KWIN_SINGLETON_FACTORY(WaylandServer)

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

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

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

void WaylandServer::destroyInternalConnection()
96
{
97
    emit terminatingInternalClientConnection();
98
    if (m_internalConnection.client) {
99 100 101 102 103 104 105 106 107
        // 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();
        }

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

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

133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
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);
148 149 150 151 152 153 154 155 156
    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
157 158 159
    if (auto menu = m_appMenuManager->appMenuForSurface(surface->surface())) {
        client->installAppMenu(menu);
    }
160 161 162
    if (auto palette = m_paletteManager->paletteForSurface(surface->surface())) {
        client->installPalette(palette);
    }
163 164 165 166 167 168 169 170 171 172
    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
173 174

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

180
bool WaylandServer::init(const QByteArray &socketName, InitalizationFlags flags)
181
{
182
    m_initFlags = flags;
183 184 185 186 187
    m_display = new KWayland::Server::Display(this);
    if (!socketName.isNull() && !socketName.isEmpty()) {
        m_display->setSocketName(QString::fromUtf8(socketName));
    }
    m_display->start();
188 189 190
    if (!m_display->isRunning()) {
        return false;
    }
191 192
    m_compositor = m_display->createCompositor(m_display);
    m_compositor->create();
193 194 195 196 197 198 199 200
    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;
            }
201 202 203 204
            if (surface->client() != xWaylandConnection()) {
                // setting surface is only relevat for Xwayland clients
                return;
            }
205 206 207 208 209 210 211 212
            auto check = [surface] (const Toplevel *t) {
                return t->surfaceId() == surface->id();
            };
            if (Toplevel *t = ws->findToplevel(check)) {
                t->setSurface(surface);
            }
        }
    );
213 214
    m_shell = m_display->createShell(m_display);
    m_shell->create();
215 216 217 218 219 220
    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
221 222 223 224 225 226 227

    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>);


228
    m_display->createShm();
229 230
    m_seat = m_display->createSeat(m_display);
    m_seat->create();
231
    m_display->createPointerGestures(PointerGesturesInterfaceVersion::UnstableV1, m_display)->create();
232
    m_display->createPointerConstraints(PointerConstraintsInterfaceVersion::UnstableV1, m_display)->create();
233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
    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());
                        }
                    }
                );
            }
        }
    );
252 253 254 255 256
    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();
257 258 259 260 261 262
    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);
263 264 265 266 267 268 269
            } else {
                m_plasmaShellSurfaces << surface;
                connect(surface, &QObject::destroyed, this,
                    [this, surface] {
                        m_plasmaShellSurfaces.removeOne(surface);
                    }
                );
270 271 272
            }
        }
    );
David Edmundson's avatar
David Edmundson committed
273 274


275 276 277 278 279 280 281 282 283
    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
284 285 286 287 288 289 290 291 292
    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);
            }
        }
    );
293 294 295 296 297 298 299 300 301
    m_paletteManager = m_display->createServerSideDecorationPaletteManager(m_display);
    m_paletteManager->create();
    connect(m_paletteManager, &ServerSideDecorationPaletteManagerInterface::paletteCreated,
        [this] (ServerSideDecorationPaletteInterface *palette) {
            if (ShellClient *client = findClient(palette->surface())) {
                client->installPalette(palette);
            }
        }
    );
David Edmundson's avatar
David Edmundson committed
302

303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
    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);
        }
    );
329 330
    auto shadowManager = m_display->createShadowManager(m_display);
    shadowManager->create();
331 332

    m_display->createDpmsManager(m_display)->create();
333 334 335 336 337 338 339 340 341 342

    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();
343 344 345 346

    m_outputManagement = m_display->createOutputManagement(m_display);
    connect(m_outputManagement, &OutputManagementInterface::configurationChangeRequested,
            this, [this](KWayland::Server::OutputConfigurationInterface *config) {
347
                kwinApp()->platform()->configurationChangeRequested(config);
348
    });
349
    m_outputManagement->create();
350 351

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

Marco Martin's avatar
Marco Martin committed
353 354 355
    m_XdgForeign = m_display->createXdgForeignInterface(m_display);
    m_XdgForeign->create();

356
    return true;
357 358
}

Marco Martin's avatar
Marco Martin committed
359 360 361 362 363
SurfaceInterface *WaylandServer::findForeignTransientForSurface(SurfaceInterface *surface)
{
    return m_XdgForeign->transientFor(surface);
}

364 365 366 367 368 369 370 371 372 373 374
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);
}

375 376 377 378 379 380 381 382 383 384 385 386 387
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
                );
            }
        );
    }
388

389
    if (hasScreenLockerIntegration()) {
390 391 392 393 394 395 396 397 398
        if (m_internalConnection.interfacesAnnounced) {
            initScreenLocker();
        } else {
            connect(m_internalConnection.registry, &KWayland::Client::Registry::interfacesAnnounced, this, &WaylandServer::initScreenLocker);
        }
    } else {
        emit initialized();
    }
}
399

400 401 402 403 404 405 406 407 408 409 410 411
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();
        }
    );
412

413 414 415
    connect(ScreenLocker::KSldApp::self(), &ScreenLocker::KSldApp::unlocked, this,
        [this] () {
            m_screenLockerClientConnection = nullptr;
416
        }
417 418 419 420
    );

    if (m_initFlags.testFlag(InitalizationFlag::LockScreen)) {
        ScreenLocker::KSldApp::self()->lock(ScreenLocker::EstablishLock::Immediate);
421
    }
422
    emit initialized();
423
}
424

425 426
void WaylandServer::initOutputs()
{
427
    if (kwinApp()->platform()->handlesOutputs()) {
428 429
        return;
    }
430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445
    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()
{
446 447 448 449
    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
450
        output->setScale(s->scale(i));
451 452
        const QRect &geo = s->geometry(i);
        output->setGlobalPosition(geo.topLeft());
453
        output->setPhysicalSize(s->physicalSize(i).toSize());
454
        output->addMode(geo.size());
455 456
        output->create();
    }
457 458
}

459
WaylandServer::SocketPairConnection WaylandServer::createConnection()
460
{
461
    SocketPairConnection ret;
462 463 464
    int sx[2];
    if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, sx) < 0) {
        qCWarning(KWIN_CORE) << "Could not create socket";
465 466 467 468 469 470 471 472 473 474 475
        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) {
476 477
        return -1;
    }
478
    m_xwayland.client = socket.connection;
479
    m_xwayland.destroyConnection = connect(m_xwayland.client, &KWayland::Server::ClientConnection::disconnected, this,
480 481 482 483
        [] {
            qFatal("Xwayland Connection died");
        }
    );
484
    return socket.fd;
485 486
}

487 488 489 490 491
void WaylandServer::destroyXWaylandConnection()
{
    if (!m_xwayland.client) {
        return;
    }
492 493 494 495
    // first terminate the clipboard sync
    if (m_xclipbaordSync.process) {
        m_xclipbaordSync.process->terminate();
    }
496 497 498 499 500
    disconnect(m_xwayland.destroyConnection);
    m_xwayland.client->destroy();
    m_xwayland.client = nullptr;
}

501 502
int WaylandServer::createInputMethodConnection()
{
503 504
    const auto socket = createConnection();
    if (!socket.connection) {
505 506
        return -1;
    }
507 508
    m_inputMethodServerConnection = socket.connection;
    return socket.fd;
509 510
}

511 512 513 514 515 516 517 518 519
void WaylandServer::destroyInputMethodConnection()
{
    if (!m_inputMethodServerConnection) {
        return;
    }
    m_inputMethodServerConnection->destroy();
    m_inputMethodServerConnection = nullptr;
}

520 521
int WaylandServer::createXclipboardSyncConnection()
{
522 523
    const auto socket = createConnection();
    if (!socket.connection) {
524 525
        return -1;
    }
526 527
    m_xclipbaordSync.client = socket.connection;
    return socket.fd;
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564
}

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));
    }
}

565 566
void WaylandServer::createInternalConnection()
{
567 568
    const auto socket = createConnection();
    if (!socket.connection) {
569 570
        return;
    }
571
    m_internalConnection.server = socket.connection;
572
    using namespace KWayland::Client;
573
    m_internalConnection.client = new ConnectionThread();
574
    m_internalConnection.client->setSocketFd(socket.fd);
575 576 577 578
    m_internalConnection.clientThread = new QThread;
    m_internalConnection.client->moveToThread(m_internalConnection.clientThread);
    m_internalConnection.clientThread->start();

579 580
    connect(m_internalConnection.client, &ConnectionThread::connected, this,
        [this] {
581 582 583 584
            Registry *registry = new Registry(this);
            EventQueue *eventQueue = new EventQueue(this);
            eventQueue->setup(m_internalConnection.client);
            registry->setEventQueue(eventQueue);
585
            registry->create(m_internalConnection.client);
586
            m_internalConnection.registry = registry;
587
            connect(registry, &Registry::shmAnnounced, this,
588
                [this] (quint32 name, quint32 version) {
589
                    m_internalConnection.shm = m_internalConnection.registry->createShmPool(name, version, this);
590 591
                }
            );
592 593 594 595 596
            connect(registry, &Registry::interfacesAnnounced, this,
                [this] {
                    m_internalConnection.interfacesAnnounced = true;
                }
            );
597 598 599 600 601 602
            registry->setup();
        }
    );
    m_internalConnection.client->initConnection();
}

603 604 605
void WaylandServer::removeClient(ShellClient *c)
{
    m_clients.removeAll(c);
606
    m_internalClients.removeAll(c);
607 608 609
    emit shellClientRemoved(c);
}

610 611 612 613 614
void WaylandServer::dispatch()
{
    if (!m_display) {
        return;
    }
615 616
    if (m_internalConnection.server) {
        m_internalConnection.server->flush();
617 618 619 620
    }
    m_display->dispatchEvents(0);
}

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

634 635 636 637 638 639 640 641 642 643 644 645 646
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;
}

647 648 649 650 651 652 653 654 655 656 657 658 659 660
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;
}

661 662 663 664 665 666 667 668 669 670 671 672 673 674
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;
}

675 676 677 678 679
AbstractClient *WaylandServer::findAbstractClient(SurfaceInterface *surface) const
{
    return findClient(surface);
}

680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695
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;
}

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 725 726 727 728 729 730 731 732 733 734 735 736 737
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;
}

738 739
bool WaylandServer::isScreenLocked() const
{
740 741 742
    if (!hasScreenLockerIntegration()) {
        return false;
    }
743 744
    return ScreenLocker::KSldApp::self()->lockState() == ScreenLocker::KSldApp::Locked ||
           ScreenLocker::KSldApp::self()->lockState() == ScreenLocker::KSldApp::AcquiringLock;
745 746
}

747 748 749 750 751
bool WaylandServer::hasScreenLockerIntegration() const
{
    return !m_initFlags.testFlag(InitalizationFlag::NoLockScreenIntegration);
}

752
}