1 files changed, 42 insertions, 0 deletions
diff --git a/Userland/Services/LaunchServer/ConnectionFromClient.h b/Userland/Services/LaunchServer/ConnectionFromClient.h
new file mode 100644
index 0000000000..3e50997802
--- /dev/null
+++ b/Userland/Services/LaunchServer/ConnectionFromClient.h
@@ -0,0 +1,42 @@
+/*
+ * Copyright (c) 2020, Nicholas Hollett <niax@niax.co.uk>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#pragma once
+
+#include <LaunchServer/LaunchClientEndpoint.h>
+#include <LaunchServer/LaunchServerEndpoint.h>
+#include <LibIPC/ConnectionFromClient.h>
+
+namespace LaunchServer {
+
+class ConnectionFromClient final : public IPC::ConnectionFromClient<LaunchClientEndpoint, LaunchServerEndpoint> {
+    C_OBJECT(ConnectionFromClient)
+public:
+    ~ConnectionFromClient() override;
+
+    virtual void die() override;
+
+private:
+    explicit ConnectionFromClient(NonnullOwnPtr<Core::Stream::LocalSocket>, int client_id);
+
+    virtual Messages::LaunchServer::OpenUrlResponse open_url(URL const&, String const&) override;
+    virtual Messages::LaunchServer::GetHandlersForUrlResponse get_handlers_for_url(URL const&) override;
+    virtual Messages::LaunchServer::GetHandlersWithDetailsForUrlResponse get_handlers_with_details_for_url(URL const&) override;
+    virtual void add_allowed_url(URL const&) override;
+    virtual void add_allowed_handler_with_any_url(String const&) override;
+    virtual void add_allowed_handler_with_only_specific_urls(String const&, Vector<URL> const&) override;
+    virtual void seal_allowlist() override;
+
+    struct AllowlistEntry {
+        String handler_name;
+        bool any_url { false };
+        Vector<URL> urls;
+    };
+
+    Vector<AllowlistEntry> m_allowlist;
+    bool m_allowlist_is_sealed { false };
+};
+}