From 8055ccf4821a6ae75d465c7dc99ec483d86dd470 Mon Sep 17 00:00:00 2001
From: N2614 <N2614@385bec56-5757-e545-9c3a-d8741f4650f1>
Date: Tue, 24 Jan 2012 04:34:06 +0000
Subject: [PATCH] =?UTF-8?q?=E3=83=97=E3=83=AA=E3=82=A4=E3=83=B3=E3=82=B9?=
 =?UTF-8?q?=E3=83=88=E3=83=BC=E3=83=AB=E3=83=80=E3=82=A6=E3=83=B3=E3=83=AD?=
 =?UTF-8?q?=E3=83=BC=E3=83=89=E5=BE=8C=E3=81=AE=E5=87=A6=E7=90=86=E3=82=92?=
 =?UTF-8?q?=E8=BF=BD=E5=8A=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

git-svn-id: file:///Volumes/Transfer/gigaleak_20231201/2020-05-23%20-%20ctr.7z%20+%20svn_v1.068.zip/ctr/svn/ctr_Repair@600 385bec56-5757-e545-9c3a-d8741f4650f1
---
 .../sources/ConsoleRestore/Controller.cpp     | 102 +++++++++++++++---
 .../sources/ConsoleRestore/Importer.cpp       |   2 +-
 2 files changed, 87 insertions(+), 17 deletions(-)

diff --git a/trunk/ConsoleDataMigration/sources/ConsoleRestore/Controller.cpp b/trunk/ConsoleDataMigration/sources/ConsoleRestore/Controller.cpp
index e18a298..191ca5e 100644
--- a/trunk/ConsoleDataMigration/sources/ConsoleRestore/Controller.cpp
+++ b/trunk/ConsoleDataMigration/sources/ConsoleRestore/Controller.cpp
@@ -101,7 +101,7 @@ bool NeedsAcAdater(common::HardwareStateManager& manager)
 
 bool HasValidRtcData()
 {
-    return !s_CheckSdOnlyMode && !s_NupOnlyMode && !s_GetIvsOnlyMode && s_DownloadPreinstallMode;
+    return !s_CheckSdOnlyMode && !s_NupOnlyMode && !s_GetIvsOnlyMode && !s_DownloadPreinstallMode;
 }
 
 typedef enum RestoreState
@@ -149,12 +149,15 @@ typedef enum RestoreState
     INITIALIZE_AND_SHUTDOWN, //　本体初期化後にシャットダウン中
     FAIL, // 失敗
 
-    PREINSTALL_WAIT_SYNC_TICKET, //プリインストールダウンロードモード・Aボタン待ち
-    PREINSTALL_WAIT_USER_SD_INSERT, // プリインストールダウンロードモード・ユーザSD挿入待ち
-    PREINSTALL_CHECK_SD, // プリインストールダウンロードモード・ユーザSDのチェック
-    PREINSTALL_CHECK_SD_FAIL, // プリインストールダウンロードモード・ユーザSDのチェック失敗
-    PREINSTALL_DOWNLOAD_APP, // プリインストールダウンロードモード・アプリダウンロード
-    PREINSTALL_WAIT_USER_SD_EJECT // プリインストールダウンロードモード・ユーザSD抜き出し待ち
+    // プリインストールダウンロードモード
+    PREINSTALL_WAIT_SYNC_TICKET, // 開始ボタン待ち
+    PREINSTALL_WAIT_USER_SD_INSERT, // ユーザSD挿入待ち
+    PREINSTALL_CHECK_USER_SD, // ユーザSDのチェック
+    PREINSTALL_CHECK_SD_FAIL, // ユーザSDのチェック失敗
+    PREINSTALL_FAIL_CHECK_REPAIR_SD, // ユーザSDのチェック失敗後の修理用SDのチェック
+    PREINSTALL_DOWNLOAD_APP, // アプリダウンロード
+    PREINSTALL_WAIT_USER_SD_EJECT, // ユーザSD抜き出し待ち
+    PREINSTALL_CHECK_REPAIR_SD // 修理用SDのチェック
 } RestoreState;
 
 #define BREAK_IF_STATE_CHANGED(saved, current) \
@@ -1636,9 +1639,9 @@ void ControlState(common::HardwareStateManager& manager, ::std::vector<std::stri
         }
         break;
 
-        case PREINSTALL_CHECK_SD:
+        case PREINSTALL_CHECK_USER_SD:
         {
-            RestoreState saved = PREINSTALL_CHECK_SD;
+            RestoreState saved = PREINSTALL_CHECK_USER_SD;
             if (nn::fs::IsSdmcInserted())
             {
                 NN_LOG("Check User's SD Card\n");
@@ -1669,13 +1672,68 @@ void ControlState(common::HardwareStateManager& manager, ::std::vector<std::stri
         case PREINSTALL_CHECK_SD_FAIL:
         {
             operationMessage.push_back(::std::string("Check User's SD Card Failed."));
-            operationMessage.push_back(::std::string("Pull Out SD Card"));
+            operationMessage.push_back(::std::string("Pull Out User's SD Card"));
+            operationMessage.push_back(::std::string("and Insert Repairing SD Card"));
+        }
+        break;
+
+        case PREINSTALL_FAIL_CHECK_REPAIR_SD:
+        {
+            RestoreState saved = PREINSTALL_FAIL_CHECK_REPAIR_SD;
+            if (nn::fs::IsSdmcInserted())
+            {
+                NN_LOG("Check SD Card\n");
+                // 修理用SDカードにはAP設定ファイルがあるはず
+                if (ExistsAPSetting())
+                {
+                    NN_LOG("AP Setting Exists. OK\n");
+                    COMMON_LOGGER("Download Preinstall failed\n");
+                    ChangeState(saved, s_RestoreState, FAIL);
+                }
+                else
+                {
+                    // SD挿抜が起こったら抜ける
+                    BREAK_IF_STATE_CHANGED(saved, s_RestoreState);
+                    ChangeState(saved, s_RestoreState, PREINSTALL_CHECK_SD_FAIL);
+                }
+            }
         }
         break;
 
         case PREINSTALL_DOWNLOAD_APP:
         {
+            COMMON_LOGGER("Download App Stub\n");
+            s_RestoreState = PREINSTALL_WAIT_USER_SD_EJECT;
+        }
+        break;
 
+        case PREINSTALL_WAIT_USER_SD_EJECT:
+        {
+            operationMessage.push_back(::std::string("Pull out User's SD Card"));
+            operationMessage.push_back(::std::string("and Insert Repairing SD Card"));
+        }
+        break;
+
+        case PREINSTALL_CHECK_REPAIR_SD:
+        {
+            RestoreState saved = PREINSTALL_CHECK_REPAIR_SD;
+            if (nn::fs::IsSdmcInserted())
+            {
+                NN_LOG("Check SD Card\n");
+                // 修理用SDカードにはAP設定ファイルがあるはず
+                if (ExistsAPSetting())
+                {
+                    NN_LOG("AP Setting Exists. OK\n");
+                    COMMON_LOGGER("Download Preinstall done\n");
+                    ChangeState(saved, s_RestoreState, WAIT_SD_EJECT);
+                }
+                else
+                {
+                    // SD挿抜が起こったら抜ける
+                    BREAK_IF_STATE_CHANGED(saved, s_RestoreState);
+                    ChangeState(saved, s_RestoreState, PREINSTALL_WAIT_USER_SD_EJECT);
+                }
+            }
         }
         break;
 
@@ -1715,19 +1773,19 @@ void OnSdEjected()
     // FAILのときは一旦電源を切らないと動かないようにしておく
     // IVSチェック時はユーザのSDカードを挿入してもらうため
     // プリインストール時はユーザのSDカードを挿入してもらうため
-    else if (s_RestoreState != FAIL && s_RestoreState != CHECK_IVS && s_RestoreState != PREINSTALL_WAIT_USER_SD_INSERT)
+    else if (s_RestoreState != FAIL && s_RestoreState != CHECK_IVS && s_RestoreState != PREINSTALL_WAIT_USER_SD_INSERT
+            && s_RestoreState != PREINSTALL_WAIT_USER_SD_EJECT && s_RestoreState != PREINSTALL_CHECK_SD_FAIL)
     {
         InitializeState();
         ClearFileReadResult();
     }
-    else if(s_RestoreState == PREINSTALL_CHECK_SD)
+    else if(s_RestoreState == PREINSTALL_CHECK_USER_SD)
     {
         s_RestoreState = PREINSTALL_WAIT_USER_SD_INSERT;
     }
-    // ユーザのSDが抜かれてからFAILにする
-    else if(s_RestoreState == PREINSTALL_CHECK_SD_FAIL)
+    else if(s_RestoreState == PREINSTALL_CHECK_REPAIR_SD)
     {
-        s_RestoreState = FAIL;
+        s_RestoreState = PREINSTALL_WAIT_USER_SD_EJECT;
     }
 }
 
@@ -1743,7 +1801,19 @@ void OnSdInserted()
     {
         // SDカードが変わるのでファイルチェックは初期化する
         common::InitializeFileCheck();
-        s_RestoreState = PREINSTALL_CHECK_SD;
+        s_RestoreState = PREINSTALL_CHECK_USER_SD;
+    }
+    else if(s_RestoreState == PREINSTALL_WAIT_USER_SD_EJECT)
+    {
+        // SDカードが変わるのでファイルチェックは初期化する
+        common::InitializeFileCheck();
+        s_RestoreState = PREINSTALL_CHECK_REPAIR_SD;
+    }
+    else if(s_RestoreState == PREINSTALL_CHECK_SD_FAIL)
+    {
+        // SDカードが変わるのでファイルチェックは初期化する
+        common::InitializeFileCheck();
+        s_RestoreState = PREINSTALL_FAIL_CHECK_REPAIR_SD;
     }
 }
 
diff --git a/trunk/ConsoleDataMigration/sources/ConsoleRestore/Importer.cpp b/trunk/ConsoleDataMigration/sources/ConsoleRestore/Importer.cpp
index 51b34bb..1b7932f 100644
--- a/trunk/ConsoleDataMigration/sources/ConsoleRestore/Importer.cpp
+++ b/trunk/ConsoleDataMigration/sources/ConsoleRestore/Importer.cpp
@@ -1448,7 +1448,7 @@ bool ReadSetting(bool* nupOnly, bool* getIvs, bool* checkSd, bool* skipNup, bool
             ReadSettingFlag(&configfileLoader, L"GET_SDCI", getIvs, "GET SDCI Mode.\n");
             ReadSettingFlag(&configfileLoader, L"CHECK_SD", checkSd, "CHECK SD Mode.\n");
             ReadSettingFlag(&configfileLoader, L"SKIP_NUP", skipNup, "Skip NUP Mode.\n");
-            ReadSettingFlag(&configfileLoader, L"DL_PREINSTALL", dlPreinstall, "Download Preinstall App.\n");
+            ReadSettingFlag(&configfileLoader, L"DL_PREINSTALL", dlPreinstall, "Download PreInstall Application Mode.\n");
 
             configfileLoader.Finalize();