summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibIPC/MultiServer.h
blob: 6efd056d11cbb70a1f5036574bdcf3d709ae5357 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/*
 * Copyright (c) 2021, Andreas Kling <kling@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/Error.h>
#include <AK/Function.h>
#include <LibCore/LocalServer.h>
#include <LibIPC/ConnectionFromClient.h>

namespace IPC {

template<typename ConnectionFromClientType>
class MultiServer {
public:
    static ErrorOr<NonnullOwnPtr<MultiServer>> try_create(Optional<DeprecatedString> socket_path = {})
    {
        auto server = TRY(Core::LocalServer::try_create());
        TRY(server->take_over_from_system_server(socket_path.value_or({})));
        return adopt_nonnull_own_or_enomem(new (nothrow) MultiServer(move(server)));
    }

    Function<void(ConnectionFromClientType&)> on_new_client;

private:
    explicit MultiServer(NonnullRefPtr<Core::LocalServer> server)
        : m_server(move(server))
    {
        m_server->on_accept = [&](auto client_socket) {
            auto client_id = ++m_next_client_id;

            auto client = IPC::new_client_connection<ConnectionFromClientType>(move(client_socket), client_id);
            if (on_new_client)
                on_new_client(*client);
        };
    }

    int m_next_client_id { 0 };
    RefPtr<Core::LocalServer> m_server;
};

}