git subrepo clone https://github.com/Atmosphere-NX/Atmosphere-libs libraries

subrepo:
  subdir:   "libraries"
  merged:   "07af583b"
upstream:
  origin:   "https://github.com/Atmosphere-NX/Atmosphere-libs"
  branch:   "master"
  commit:   "07af583b"
git-subrepo:
  version:  "0.4.0"
  origin:   "https://github.com/ingydotnet/git-subrepo"
  commit:   "5d6aba9"

libraries/libstratosphere/source/sm/sm_ams.c

@@ -0,0 +1,131 @@
+/*
+ * Copyright (c) 2018-2019 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/>.
+ */
+#define NX_SERVICE_ASSUME_NON_DOMAIN
+#include "../service_guard.h"
+#include "sm_ams.h"
+
+static Result _smAtmosphereCmdHas(bool *out, SmServiceName name, u32 cmd_id) {
+    u8 tmp;
+    Result rc = serviceDispatchInOut(smGetServiceSession(), cmd_id, name, tmp);
+    if (R_SUCCEEDED(rc) && out) *out = tmp & 1;
+    return rc;
+}
+
+static Result _smAtmosphereCmdInServiceNameNoOut(SmServiceName name, Service *srv, u32 cmd_id) {
+    return serviceDispatchIn(srv, cmd_id, name);
+}
+
+Result smAtmosphereHasService(bool *out, SmServiceName name) {
+    return _smAtmosphereCmdHas(out, name, 65100);
+}
+
+Result smAtmosphereWaitService(SmServiceName name) {
+    return _smAtmosphereCmdInServiceNameNoOut(name, smGetServiceSession(), 65101);
+}
+
+Result smAtmosphereHasMitm(bool *out, SmServiceName name) {
+    return _smAtmosphereCmdHas(out, name, 65004);
+}
+
+Result smAtmosphereWaitMitm(SmServiceName name) {
+    return _smAtmosphereCmdInServiceNameNoOut(name, smGetServiceSession(), 65005);
+}
+
+static Service g_smAtmosphereMitmSrv;
+
+NX_GENERATE_SERVICE_GUARD(smAtmosphereMitm);
+
+Result _smAtmosphereMitmInitialize(void) {
+    return smAtmosphereOpenSession(&g_smAtmosphereMitmSrv);
+}
+
+void _smAtmosphereMitmCleanup(void) {
+    smAtmosphereCloseSession(&g_smAtmosphereMitmSrv);
+}
+
+Service* smAtmosphereMitmGetServiceSession(void) {
+    return &g_smAtmosphereMitmSrv;
+}
+
+Result smAtmosphereOpenSession(Service *out) {
+    Handle sm_handle;
+    Result rc = svcConnectToNamedPort(&sm_handle, "sm:");
+    while (R_VALUE(rc) == KERNELRESULT(NotFound)) {
+        svcSleepThread(50000000ul);
+        rc = svcConnectToNamedPort(&sm_handle, "sm:");
+    }
+
+    if (R_SUCCEEDED(rc)) {
+        serviceCreate(out, sm_handle);
+    }
+
+    if (R_SUCCEEDED(rc)) {
+        const u64 pid_placeholder = 0;
+        rc = serviceDispatchIn(out, 0, pid_placeholder, .in_send_pid = true);
+    }
+
+    return rc;
+}
+
+void smAtmosphereCloseSession(Service *srv) {
+    serviceClose(srv);
+}
+
+Result smAtmosphereMitmInstall(Service *fwd_srv, Handle *handle_out, Handle *query_out, SmServiceName name) {
+    Handle tmp_handles[2];
+    Result rc = serviceDispatchIn(fwd_srv, 65000, name,
+        .out_handle_attrs = { SfOutHandleAttr_HipcMove, SfOutHandleAttr_HipcMove },
+        .out_handles = tmp_handles,
+    );
+
+    if (R_SUCCEEDED(rc)) {
+        *handle_out = tmp_handles[0];
+        *query_out  = tmp_handles[1];
+    }
+
+    return rc;
+}
+
+Result smAtmosphereMitmUninstall(SmServiceName name) {
+    return _smAtmosphereCmdInServiceNameNoOut(name, smGetServiceSession(), 65001);
+}
+
+Result smAtmosphereMitmDeclareFuture(SmServiceName name) {
+    return _smAtmosphereCmdInServiceNameNoOut(name, smGetServiceSession(), 65006);
+}
+
+Result smAtmosphereMitmAcknowledgeSession(Service *srv_out, void *_out, SmServiceName name) {
+    struct {
+        u64 process_id;
+        u64 program_id;
+        u64 keys_held;
+        u64 flags;
+    } *out = _out;
+    _Static_assert(sizeof(*out) == 0x20, "sizeof(*out) == 0x20");
+
+    Handle tmp_handle;
+
+    Result rc = serviceDispatchInOut(&g_smAtmosphereMitmSrv, 65003, name, *out,
+        .out_handle_attrs = { SfOutHandleAttr_HipcMove },
+        .out_handles = &tmp_handle,
+    );
+
+    if (R_SUCCEEDED(rc)) {
+        serviceCreate(srv_out, tmp_handle);
+    }
+
+    return rc;
+}

libraries/libstratosphere/source/sm/sm_ams.h

@@ -0,0 +1,35 @@
+/**
+ * @file sm_ams.h
+ * @brief Service manager (sm) IPC wrapper for Atmosphere extensions.
+ * @author SciresM
+ * @copyright libnx Authors
+ */
+#pragma once
+#include <switch/types.h>
+#include <switch/kernel/event.h>
+#include <switch/services/sm.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+Result smAtmosphereHasService(bool *out, SmServiceName name);
+Result smAtmosphereWaitService(SmServiceName name);
+Result smAtmosphereHasMitm(bool *out, SmServiceName name);
+Result smAtmosphereWaitMitm(SmServiceName name);
+
+Result smAtmosphereMitmInitialize(void);
+void smAtmosphereMitmExit(void);
+Service *smAtmosphereMitmGetServiceSession();
+
+Result smAtmosphereOpenSession(Service *out);
+void smAtmosphereCloseSession(Service *srv);
+
+Result smAtmosphereMitmInstall(Service *fwd_srv, Handle *handle_out, Handle *query_out, SmServiceName name);
+Result smAtmosphereMitmUninstall(SmServiceName name);
+Result smAtmosphereMitmDeclareFuture(SmServiceName name);
+Result smAtmosphereMitmAcknowledgeSession(Service *srv_out, void *info_out, SmServiceName name);
+
+#ifdef __cplusplus
+}
+#endif

libraries/libstratosphere/source/sm/sm_api.cpp

@@ -0,0 +1,63 @@
+/*
+ * Copyright (c) 2018-2019 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 "sm_utils.hpp"
+
+namespace ams::sm {
+
+    /* Ordinary SM API. */
+    Result GetService(Service *out, ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smGetServiceWrapper(out, impl::ConvertName(name));
+        });
+    }
+
+    Result RegisterService(Handle *out, ServiceName name, size_t max_sessions, bool is_light) {
+        return impl::DoWithUserSession([&]() {
+            return smRegisterService(out, impl::ConvertName(name), is_light, static_cast<int>(max_sessions));
+        });
+    }
+
+    Result UnregisterService(ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smUnregisterService(impl::ConvertName(name));
+        });
+    }
+
+    /* Atmosphere extensions. */
+    Result HasService(bool *out, ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smAtmosphereHasService(out, impl::ConvertName(name));
+        });
+    }
+
+    Result WaitService(ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smAtmosphereWaitService(impl::ConvertName(name));
+        });
+    }
+
+    namespace impl {
+
+        void DoWithSessionImpl(void (*Invoker)(void *), void *Function) {
+            impl::DoWithUserSession([&]() {
+                Invoker(Function);
+                return ResultSuccess();
+            });
+        }
+
+    }
+
+}

libraries/libstratosphere/source/sm/sm_manager_api.cpp

@@ -0,0 +1,40 @@
+/*
+ * Copyright (c) 2018-2019 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 "sm_utils.hpp"
+#include "smm_ams.h"
+
+namespace ams::sm::manager {
+
+    /* Manager API. */
+    Result RegisterProcess(os::ProcessId process_id, ncm::ProgramId program_id, cfg::OverrideStatus status, const void *acid, size_t acid_size, const void *aci, size_t aci_size) {
+        static_assert(sizeof(status) == sizeof(CfgOverrideStatus), "CfgOverrideStatus definition");
+        return smManagerAtmosphereRegisterProcess(static_cast<u64>(process_id), static_cast<u64>(program_id), reinterpret_cast<const CfgOverrideStatus *>(&status), acid, acid_size, aci, aci_size);
+    }
+
+    Result UnregisterProcess(os::ProcessId process_id) {
+        return smManagerUnregisterProcess(static_cast<u64>(process_id));
+    }
+
+    /* Atmosphere extensions. */
+    Result EndInitialDefers() {
+        return smManagerAtmosphereEndInitialDefers();
+    }
+
+    Result HasMitm(bool *out, ServiceName name) {
+        return smManagerAtmosphereHasMitm(out, impl::ConvertName(name));
+    }
+
+}

libraries/libstratosphere/source/sm/sm_mitm_api.cpp

@@ -0,0 +1,57 @@
+/*
+ * Copyright (c) 2018-2019 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 "sm_utils.hpp"
+
+namespace ams::sm::mitm {
+
+    /* Mitm API. */
+    Result InstallMitm(Handle *out_port, Handle *out_query, ServiceName name) {
+        return impl::DoWithPerThreadSession([&](Service *fwd) {
+            return smAtmosphereMitmInstall(fwd, out_port, out_query, impl::ConvertName(name));
+        });
+    }
+
+    Result UninstallMitm(ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smAtmosphereMitmUninstall(impl::ConvertName(name));
+        });
+    }
+
+    Result DeclareFutureMitm(ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smAtmosphereMitmDeclareFuture(impl::ConvertName(name));
+        });
+    }
+
+    Result AcknowledgeSession(Service *out_service, MitmProcessInfo *out_info, ServiceName name) {
+        return impl::DoWithMitmAcknowledgementSession([&]() {
+            return smAtmosphereMitmAcknowledgeSession(out_service, reinterpret_cast<void *>(out_info), impl::ConvertName(name));
+        });
+    }
+
+    Result HasMitm(bool *out, ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smAtmosphereHasMitm(out, impl::ConvertName(name));
+        });
+    }
+
+    Result WaitMitm(ServiceName name) {
+        return impl::DoWithUserSession([&]() {
+            return smAtmosphereWaitMitm(impl::ConvertName(name));
+        });
+    }
+
+}

libraries/libstratosphere/source/sm/sm_utils.cpp

@@ -0,0 +1,42 @@
+/*
+ * Copyright (c) 2018-2019 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 "sm_utils.hpp"
+
+namespace ams::sm::impl {
+
+    namespace {
+
+        /* Globals. */
+        os::RecursiveMutex g_user_session_mutex;
+        os::RecursiveMutex g_mitm_ack_session_mutex;
+        os::RecursiveMutex g_per_thread_session_mutex;
+
+    }
+
+    /* Utilities. */
+    os::RecursiveMutex &GetUserSessionMutex() {
+        return g_user_session_mutex;
+    }
+
+    os::RecursiveMutex &GetMitmAcknowledgementSessionMutex() {
+        return g_mitm_ack_session_mutex;
+    }
+
+    os::RecursiveMutex &GetPerThreadSessionMutex() {
+        return g_per_thread_session_mutex;
+    }
+
+}

libraries/libstratosphere/source/sm/sm_utils.hpp

@@ -0,0 +1,70 @@
+/*
+ * Copyright (c) 2018-2019 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>
+#include "sm_ams.h"
+
+namespace ams::sm::impl {
+
+    /* Utilities. */
+    os::RecursiveMutex &GetUserSessionMutex();
+    os::RecursiveMutex &GetMitmAcknowledgementSessionMutex();
+    os::RecursiveMutex &GetPerThreadSessionMutex();
+
+    template<typename F>
+    Result DoWithUserSession(F f) {
+        std::scoped_lock<os::RecursiveMutex &> lk(GetUserSessionMutex());
+        {
+            R_ASSERT(smInitialize());
+            ON_SCOPE_EXIT { smExit(); };
+
+            return f();
+        }
+    }
+
+    template<typename F>
+    Result DoWithMitmAcknowledgementSession(F f) {
+        std::scoped_lock<os::RecursiveMutex &> lk(GetMitmAcknowledgementSessionMutex());
+        {
+            R_ASSERT(smAtmosphereMitmInitialize());
+            ON_SCOPE_EXIT { smAtmosphereMitmExit(); };
+
+            return f();
+        }
+    }
+
+    template<typename F>
+    Result DoWithPerThreadSession(F f) {
+        Service srv;
+        {
+            std::scoped_lock<os::RecursiveMutex &> lk(GetPerThreadSessionMutex());
+            R_ASSERT(smAtmosphereOpenSession(&srv));
+        }
+        {
+            ON_SCOPE_EXIT { smAtmosphereCloseSession(&srv); };
+            return f(&srv);
+        }
+    }
+
+    NX_CONSTEXPR SmServiceName ConvertName(sm::ServiceName name) {
+        static_assert(sizeof(SmServiceName) == sizeof(sm::ServiceName));
+        SmServiceName ret = {};
+        __builtin_memcpy(&ret, &name, sizeof(sm::ServiceName));
+        return ret;
+    }
+
+}

libraries/libstratosphere/source/sm/smm_ams.c

@@ -0,0 +1,49 @@
+/*
+ * Copyright (c) 2018-2019 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 "smm_ams.h"
+
+Result smManagerAtmosphereEndInitialDefers(void) {
+    return serviceDispatch(smManagerGetServiceSession(), 65000);
+}
+
+Result smManagerAtmosphereRegisterProcess(u64 pid, u64 tid, const CfgOverrideStatus *status, const void *acid_sac, size_t acid_sac_size, const void *aci_sac, size_t aci_sac_size) {
+    const struct {
+        u64 pid;
+        u64 tid;
+        CfgOverrideStatus status;
+    } in = { pid, tid, *status };
+    return serviceDispatchIn(smManagerGetServiceSession(), 65002, in,
+        .buffer_attrs = {
+            SfBufferAttr_In | SfBufferAttr_HipcMapAlias,
+            SfBufferAttr_In | SfBufferAttr_HipcMapAlias,
+        },
+        .buffers = {
+            { acid_sac, acid_sac_size },
+            { aci_sac, aci_sac_size },
+        },
+    );
+}
+
+static Result _smManagerAtmosphereCmdHas(bool *out, SmServiceName name, u32 cmd_id) {
+    u8 tmp;
+    Result rc = serviceDispatchInOut(smManagerGetServiceSession(), cmd_id, name, tmp);
+    if (R_SUCCEEDED(rc) && out) *out = tmp & 1;
+    return rc;
+}
+
+Result smManagerAtmosphereHasMitm(bool *out, SmServiceName name) {
+    return _smManagerAtmosphereCmdHas(out, name, 65001);
+}

libraries/libstratosphere/source/sm/smm_ams.h

@@ -0,0 +1,25 @@
+/**
+ * @file smm_ams.h
+ * @brief Service manager manager (sm:m) IPC wrapper for Atmosphere extensions.
+ * @author SciresM
+ * @copyright libnx Authors
+ */
+#pragma once
+#include <switch.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+typedef struct {
+    u64 keys_held;
+    u64 flags;
+} CfgOverrideStatus;
+
+Result smManagerAtmosphereEndInitialDefers(void);
+Result smManagerAtmosphereRegisterProcess(u64 pid, u64 tid, const CfgOverrideStatus *status, const void *acid_sac, size_t acid_sac_size, const void *aci_sac, size_t aci_sac_size);
+Result smManagerAtmosphereHasMitm(bool *out, SmServiceName name);
+
+#ifdef __cplusplus
+}
+#endif