exo2: implement SmcSetKernelCarveoutRegion

exosphere2/program/source/smc/secmon_smc_aes.cpp

@@ -111,7 +111,7 @@ namespace ams::secmon::smc {
             u8 kek_source[AesKeySize];
             std::memcpy(kek_source, std::addressof(args.r[1]), AesKeySize);
 
-            const int generation = std::min<int>(args.r[3] - 1, pkg1::KeyGeneration_1_0_0);
+            const int generation = std::max<int>(static_cast<int>(args.r[3]) - 1, pkg1::KeyGeneration_1_0_0);
 
             const util::BitPack32 option = { static_cast<u32>(args.r[4]) };
             const bool is_device_unique  = option.Get<GenerateAesKekOption::IsDeviceUnique>();

exosphere2/program/source/smc/secmon_smc_carveout.cpp

@@ -15,13 +15,27 @@
  */
 #include <exosphere.hpp>
 #include "../secmon_error.hpp"
+#include "../secmon_setup.hpp"
 #include "secmon_smc_carveout.hpp"
 
 namespace ams::secmon::smc {
 
     SmcResult SmcSetKernelCarveoutRegion(SmcArguments &args) {
-        /* TODO */
-        return SmcResult::NotImplemented;
+        /* Decode arguments. */
+        const int       index   = args.r[1];
+        const uintptr_t address = args.r[2];
+        const size_t    size    = args.r[3];
+
+        /* Validate arguments. */
+        SMC_R_UNLESS(0 <= index && index < KernelCarveoutCount, InvalidArgument);
+        SMC_R_UNLESS(util::IsAligned(address, 128_KB),          InvalidArgument);
+        SMC_R_UNLESS(util::IsAligned(size,    128_KB),          InvalidArgument);
+        SMC_R_UNLESS(size <= CarveoutSizeMax,                   InvalidArgument);
+
+        /* Set the carveout. */
+        SetKernelCarveoutRegion(index, address, size);
+
+        return SmcResult::Success;
     }
 
 }

exosphere2/program/source/smc/secmon_smc_handler.cpp

@@ -242,7 +242,7 @@ namespace ams::secmon::smc {
 
             util::WaitMicroSeconds(1000);
         }
-        if (args.r[0] != static_cast<u64>(SmcResult::Success)) {
+        if (args.r[0] != static_cast<u64>(SmcResult::Success) && info.function_id != 0xC3000007 /* generate aes key fails during SetupKekAccessKeys */) {
             *(volatile u32 *)(MemoryRegionVirtualDebug.GetAddress())            = 0xCCCCCCCC;
             *(volatile u32 *)(MemoryRegionVirtualDebug.GetAddress() + 0x10)     = static_cast<u32>(info.function_id);
             for (size_t i = 0; i < sizeof(args) / sizeof(u32); ++i) {