/*
 * Copyright (c) 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/>.
 */
#include <stratosphere.hpp>

#include "spl_random_service.hpp"
#include "spl_general_service.hpp"
#include "spl_crypto_service.hpp"
#include "spl_ssl_service.hpp"
#include "spl_es_service.hpp"
#include "spl_fs_service.hpp"
#include "spl_manu_service.hpp"

#include "spl_deprecated_service.hpp"

namespace ams {

    namespace spl {

        namespace {

            struct SplServerOptions {
                static constexpr size_t PointerBufferSize   = 0x800;
                static constexpr size_t MaxDomains          = 0;
                static constexpr size_t MaxDomainObjects    = 0;
                static constexpr bool CanDeferInvokeRequest = false;
                static constexpr bool CanManageMitmServers  = false;
            };

            enum PortIndex {
                PortIndex_General,
                PortIndex_Random,
                PortIndex_Crypto,
                PortIndex_Fs,
                PortIndex_Ssl,
                PortIndex_Es,
                PortIndex_Manu,
                PortIndex_Count,
            };

            constexpr sm::ServiceName RandomServiceName = sm::ServiceName::Encode("csrng");
            constexpr size_t          RandomMaxSessions = 10; /* NOTE: Official is 9. */

            constexpr sm::ServiceName GeneralServiceName = sm::ServiceName::Encode("spl:");
            constexpr size_t          DeprecatedMaxSessions = 13;
            constexpr size_t          GeneralMaxSessions = 9; /* NOTE: Official is 8. */

            constexpr sm::ServiceName CryptoServiceName = sm::ServiceName::Encode("spl:mig");
            constexpr size_t          CryptoMaxSessions = 7; /* NOTE: Official is 6. */

            constexpr sm::ServiceName SslServiceName = sm::ServiceName::Encode("spl:ssl");
            constexpr size_t          SslMaxSessions = 2; /* NOTE: Official is 2. */

            constexpr sm::ServiceName EsServiceName = sm::ServiceName::Encode("spl:es");
            constexpr size_t          EsMaxSessions = 2; /* NOTE: Official is 2. */

            constexpr sm::ServiceName FsServiceName = sm::ServiceName::Encode("spl:fs");
            constexpr size_t          FsMaxSessions = 3; /* NOTE: Official is 1. */

            constexpr sm::ServiceName ManuServiceName = sm::ServiceName::Encode("spl:manu");
            constexpr size_t          ManuMaxSessions = 1; /* NOTE: Official is 1. */

            /* csrng, spl:, spl:mig, spl:ssl, spl:es, spl:fs, spl:manu. */
            /* TODO: Consider max sessions enforcement? */
            constexpr size_t ModernMaxSessions = GeneralMaxSessions + CryptoMaxSessions + SslMaxSessions + EsMaxSessions + FsMaxSessions + ManuMaxSessions;
            constexpr size_t NumSessions = RandomMaxSessions + std::max(DeprecatedMaxSessions, ModernMaxSessions) + 1;

            class ServerManager final : public sf::hipc::ServerManager<PortIndex_Count, SplServerOptions, NumSessions> {
                private:
                    sf::ExpHeapAllocator *m_allocator;
                    spl::SecureMonitorManager *m_secure_monitor_manager;
                    spl::GeneralService m_general_service;
                    sf::UnmanagedServiceObjectByPointer<spl::impl::IGeneralInterface, spl::GeneralService> m_general_service_object;
                    spl::RandomService m_random_service;
                    sf::UnmanagedServiceObjectByPointer<spl::impl::IRandomInterface, spl::RandomService> m_random_service_object;
                public:
                    ServerManager(sf::ExpHeapAllocator *allocator, spl::SecureMonitorManager *manager) : m_allocator(allocator), m_secure_monitor_manager(manager), m_general_service(manager), m_general_service_object(std::addressof(m_general_service)), m_random_service(manager), m_random_service_object(std::addressof(m_random_service)) {
                        /* ... */
                    }
                private:
                    virtual ams::Result OnNeedsToAccept(int port_index, Server *server) override;
            };

            using Allocator     = sf::ExpHeapAllocator;
            using ObjectFactory = sf::ObjectFactory<sf::ExpHeapAllocator::Policy>;

            alignas(0x40) constinit u8 g_server_allocator_buffer[8_KB];
            Allocator g_server_allocator;
            constinit SecureMonitorManager g_secure_monitor_manager;

            constinit bool g_use_new_server = false;

            ServerManager g_server_manager(std::addressof(g_server_allocator), std::addressof(g_secure_monitor_manager));

            ams::Result ServerManager::OnNeedsToAccept(int port_index, Server *server) {
                switch (port_index) {
                    case PortIndex_General:
                        if (g_use_new_server) {
                            R_RETURN(this->AcceptImpl(server, m_general_service_object.GetShared()));
                        } else {
                            R_RETURN(this->AcceptImpl(server, ObjectFactory::CreateSharedEmplaced<spl::impl::IDeprecatedGeneralInterface, spl::DeprecatedService>(m_allocator, m_secure_monitor_manager)));
                        }
                    case PortIndex_Random:
                        R_RETURN(this->AcceptImpl(server, m_random_service_object.GetShared()));
                    case PortIndex_Crypto:
                        R_RETURN(this->AcceptImpl(server, ObjectFactory::CreateSharedEmplaced<spl::impl::ICryptoInterface, spl::CryptoService>(m_allocator, m_secure_monitor_manager)));
                    case PortIndex_Fs:
                        R_RETURN(this->AcceptImpl(server, ObjectFactory::CreateSharedEmplaced<spl::impl::IFsInterface,     spl::FsService>(m_allocator, m_secure_monitor_manager)));
                    case PortIndex_Ssl:
                        R_RETURN(this->AcceptImpl(server, ObjectFactory::CreateSharedEmplaced<spl::impl::ISslInterface,    spl::SslService>(m_allocator, m_secure_monitor_manager)));
                    case PortIndex_Es:
                        R_RETURN(this->AcceptImpl(server, ObjectFactory::CreateSharedEmplaced<spl::impl::IEsInterface,     spl::EsService>(m_allocator, m_secure_monitor_manager)));
                    case PortIndex_Manu:
                        R_RETURN(this->AcceptImpl(server, ObjectFactory::CreateSharedEmplaced<spl::impl::IManuInterface,   spl::ManuService>(m_allocator, m_secure_monitor_manager)));
                    AMS_UNREACHABLE_DEFAULT_CASE();
                }
            }

            void SplMain() {
                /* Setup server allocator. */
                g_server_allocator.Attach(lmem::CreateExpHeap(g_server_allocator_buffer, sizeof(g_server_allocator_buffer), lmem::CreateOption_None));

                /* Initialize secure monitor manager. */
                g_secure_monitor_manager.Initialize();

                g_use_new_server = hos::GetVersion() >= hos::Version_4_0_0;

                /* Create services. */
                const auto fw_ver = hos::GetVersion();
                R_ABORT_UNLESS(g_server_manager.RegisterServer(PortIndex_General, GeneralServiceName, fw_ver >= hos::Version_4_0_0 ? GeneralMaxSessions : DeprecatedMaxSessions));
                R_ABORT_UNLESS(g_server_manager.RegisterServer(PortIndex_Random, RandomServiceName, RandomMaxSessions));
                if (fw_ver >= hos::Version_4_0_0) {
                    R_ABORT_UNLESS(g_server_manager.RegisterServer(PortIndex_Crypto, CryptoServiceName, CryptoMaxSessions));
                    R_ABORT_UNLESS(g_server_manager.RegisterServer(PortIndex_Fs,     FsServiceName,     FsMaxSessions));
                    R_ABORT_UNLESS(g_server_manager.RegisterServer(PortIndex_Ssl,    SslServiceName,    SslMaxSessions));
                    R_ABORT_UNLESS(g_server_manager.RegisterServer(PortIndex_Es,     EsServiceName,     EsMaxSessions));
                    if (fw_ver >= hos::Version_5_0_0) {
                        g_server_manager.RegisterServer(PortIndex_Manu, ManuServiceName, ManuMaxSessions);
                    }
                }

                /* Loop forever, servicing our services. */
                g_server_manager.LoopProcess();
            }

        }

    }

    namespace init {

        void InitializeSystemModule() {
            /* Initialize our connection to sm. */
            R_ABORT_UNLESS(sm::Initialize());
        }

        void FinalizeSystemModule() { /* ... */ }

        void Startup() { /* ... */ }

    }

    void NORETURN Exit(int rc) {
        AMS_UNUSED(rc);
        AMS_ABORT("Exit called by immortal process");
    }

    void Main() {
        /* Set thread name. */
        os::SetThreadNamePointer(os::GetCurrentThread(), AMS_GET_SYSTEM_THREAD_NAME(spl, Main));
        AMS_ASSERT(os::GetThreadPriority(os::GetCurrentThread()) == AMS_GET_SYSTEM_THREAD_PRIORITY(spl, Main));

        /* Invoke SPL main. */
        spl::SplMain();

        /* This can never be reached. */
        AMS_ASSUME(false);
    }

}