exo2: implement remainder of warmboot tz code

exosphere2/program/source/smc/secmon_smc_power_management.cpp

@@ -19,6 +19,12 @@
 
 namespace ams::secmon::smc {
 
+    namespace {
+
+        constinit bool g_charger_hi_z_mode_enabled = false;
+
+    }
+
     SmcResult SmcPowerOffCpu(const SmcArguments &args) {
         /* TODO */
         return SmcResult::NotImplemented;
@@ -34,4 +40,12 @@ namespace ams::secmon::smc {
         return SmcResult::NotImplemented;
     }
 
+    bool IsChargerHiZModeEnabled() {
+        return g_charger_hi_z_mode_enabled;
+    }
+
+    void SetChargerHiZModeEnabled(bool en) {
+        g_charger_hi_z_mode_enabled = en;
+    }
+
 }

exosphere2/program/source/smc/secmon_smc_power_management.hpp

@@ -24,4 +24,7 @@ namespace ams::secmon::smc {
 
     SmcResult SmcSuspendCpu(const SmcArguments &args);
 
+    bool IsChargerHiZModeEnabled();
+    void SetChargerHiZModeEnabled(bool en);
+
 }

exosphere2/program/source/smc/secmon_smc_se_lock.cpp

@@ -0,0 +1,41 @@
+/*
+ * 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;
+    }
+
+}

exosphere2/program/source/smc/secmon_smc_se_lock.hpp

@@ -0,0 +1,26 @@
+/*
+ * 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 <exosphere.hpp>
+#include "secmon_smc_common.hpp"
+
+namespace ams::secmon::smc {
+
+    bool TryLockSecurityEngine();
+    void UnlockSecurityEngine();
+    bool IsSecurityEngineLocked();
+
+}