From 878ac59aae64227ab019d3d57de5cad4396eb61c Mon Sep 17 00:00:00 2001
From: Michael Scire <SciresM@gmail.com>
Date: Wed, 14 Nov 2018 19:49:12 -0800
Subject: [PATCH] fs.mitm: skeleton logic for protecting autorcm.

---
 .../fs_mitm/source/fsmitm_boot0storage.hpp    | 23 ++++++++++++++-----
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/stratosphere/fs_mitm/source/fsmitm_boot0storage.hpp b/stratosphere/fs_mitm/source/fsmitm_boot0storage.hpp
index d06351175..a21365ca1 100644
--- a/stratosphere/fs_mitm/source/fsmitm_boot0storage.hpp
+++ b/stratosphere/fs_mitm/source/fsmitm_boot0storage.hpp
@@ -125,16 +125,21 @@ class SectoredProxyStorage : public ProxyStorage {
 class Boot0Storage : public SectoredProxyStorage<0x200> {
     using Base = SectoredProxyStorage<0x200>;
     private:
-        bool allow_writes;
+        u64 title_id;
     private:
         HosMutex *GetMutex() {
             static HosMutex s_boot0_mutex;
             return &s_boot0_mutex;
         }
+        bool AllowWrites() {
+            return title_id < 0x0100000000001000ULL;
+        }
+        bool CanModifyBctPubks() {
+            return title_id != 0x010000000000001FULL;
+        }
     public:
-        Boot0Storage(FsStorage *s, bool w) : Base(s), allow_writes(w) { }
-        Boot0Storage(FsStorage s, bool w) : Base(s), allow_writes(w) { }
-        
+        Boot0Storage(FsStorage *s, u64 t) : Base(s), title_id(t) { }
+        Boot0Storage(FsStorage s, u64 t) : Base(s), title_id(t) { }
     public:
         virtual Result Read(void *_buffer, size_t size, u64 offset) override {
             GetMutex()->Lock();
@@ -147,10 +152,16 @@ class Boot0Storage : public SectoredProxyStorage<0x200> {
             GetMutex()->Lock();
             ON_SCOPE_EXIT { GetMutex()->Unlock(); };
             
-            if (!this->allow_writes) {
+            if (!AllowWrites()) {
                 return 0x313802;
             }
             
-            return Base::Write(_buffer, size, offset);
+            /* We care about protecting autorcm from NS. */
+            if (CanModifyBctPubks()) {
+                return Base::Write(_buffer, size, offset);
+            }
+            
+            /* TODO */
+            return 0x313802;
         }
 };
\ No newline at end of file