Newer
Older
Atmosphere / libraries / libstratosphere / source / pgl / srv / pgl_srv_shell_event_observer.hpp
/*
 * Copyright (c) 2018-2020 Atmosphère-NX
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope 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/>.
 */
#pragma once
#include <stratosphere.hpp>

namespace ams::pgl::srv {

    class IShellEventObserver {
        public:
            virtual void Notify(const pm::ProcessEventInfo &info) = 0;
    };

    class ShellEventObserverHolder : public util::IntrusiveListBaseNode<ShellEventObserverHolder> {
        private:
            IShellEventObserver *observer;
        public:
            explicit ShellEventObserverHolder(IShellEventObserver *observer) : observer(observer) { /* ... */ }

            void Notify(const pm::ProcessEventInfo &info) {
                this->observer->Notify(info);
            }
    };

    class ShellEventObserver final : public IShellEventObserver {
        private:
            static constexpr size_t QueueCapacity = 0x20;
        private:
            os::MessageQueue message_queue;
            uintptr_t queue_buffer[QueueCapacity];
            os::SystemEvent event;
            TYPED_STORAGE(lmem::HeapCommonHead) heap_head;
            lmem::HeapHandle heap_handle;
            pm::ProcessEventInfo event_info_data[QueueCapacity];
            TYPED_STORAGE(ShellEventObserverHolder) holder;
        public:
            ShellEventObserver();
            ~ShellEventObserver();

            os::SystemEvent &GetEvent() {
                return this->event;
            }

            Result PopEventInfo(pm::ProcessEventInfo *out);

            virtual void Notify(const pm::ProcessEventInfo &info) override final;
    };

    class EventObserverInterface final : public pgl::sf::IEventObserver {
        private:
            TYPED_STORAGE(ShellEventObserver) observer;
        public:
            EventObserverInterface() {
                std::memset(std::addressof(this->observer), 0, sizeof(this->observer));
                new (GetPointer(this->observer)) ShellEventObserver;
            }

            ~EventObserverInterface() {
                GetReference(this->observer).~ShellEventObserver();
            }
        public:
            virtual Result GetProcessEventHandle(ams::sf::OutCopyHandle out) override final;
            virtual Result GetProcessEventInfo(ams::sf::Out<pm::ProcessEventInfo> out) override final;
    };

}