exo2: rename exosphere2 -> exosphere

exosphere/program/source/smc/secmon_smc_se_lock.cpp

@@ -0,0 +1,70 @@
+/*
+ * 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/>.
+ */
+#include <exosphere.hpp>
+#include "../secmon_error.hpp"
+#include "secmon_smc_se_lock.hpp"
+
+namespace ams::secmon::smc {
+
+    namespace {
+
+        constinit std::atomic_bool g_is_locked = false;
+
+    }
+
+    bool TryLockSecurityEngine() {
+        bool value = false;
+        return g_is_locked.compare_exchange_strong(value, true);
+    }
+
+    void UnlockSecurityEngine() {
+        g_is_locked = false;
+    }
+
+    bool IsSecurityEngineLocked() {
+        return g_is_locked;
+    }
+
+    SmcResult LockSecurityEngineAndInvoke(SmcArguments &args, SmcHandler impl) {
+        /* Try to lock the security engine. */
+        SMC_R_UNLESS(TryLockSecurityEngine(), Busy);
+        ON_SCOPE_EXIT { UnlockSecurityEngine(); };
+
+        return impl(args);
+    }
+
+    SmcResult LockSecurityEngineAndInvokeAsync(SmcArguments &args, SmcHandler impl, GetResultHandler result_handler) {
+        /* Try to lock the security engine. */
+        SMC_R_UNLESS(TryLockSecurityEngine(), Busy);
+        auto se_guard = SCOPE_GUARD { UnlockSecurityEngine(); };
+
+        /* Try to start an async operation. */
+        const u64 async_key = BeginAsyncOperation(result_handler);
+        SMC_R_UNLESS(async_key != InvalidAsyncKey, Busy);
+        auto async_guard = SCOPE_GUARD { CancelAsyncOperation(async_key); };
+
+        /* Try to invoke the operation. */
+        SMC_R_TRY(impl(args));
+
+        /* We succeeded! Cancel our guards, and return the async key to our caller. */
+        async_guard.Cancel();
+        se_guard.Cancel();
+
+        args.r[1] = async_key;
+        return SmcResult::Success;
+    }
+
+}