Backup config files before updating them just to be safe.

jakemason · Sep 5, 2022 · 07af259 · 07af259
1 parent 5217c6e
commit 07af259
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/headers/AppData.hpp b/src/headers/AppData.hpp
@@ -203,9 +203,18 @@ class AppData
     resetEntries();
   }
 
+  void backupConfigFile()
+  {
+    std::ifstream src(SAVE_FILE_NAME, std::ios::binary);
+    std::ofstream dst(SAVE_FILE_NAME ".backup", std::ios::binary);
+
+    dst << src.rdbuf();
+  }
+
   void updateSaveFileFormat()
   {
     WARN("Updating to the new Save File Format");
+    backupConfigFile();
     std::ifstream in;
     in.open("./" SAVE_FILE_NAME);
     if (!in)
@@ -220,6 +229,8 @@ class AppData
     bool isMultiline   = false;
     bool isHiddenField = false;
 
+    int abortCounter = 0;
+    int abortLimit   = 10000;
     while (!in.eof())
     {
       in >> isMultiline;
@@ -236,6 +247,14 @@ class AppData
         strcpy(toAdd.expandsTo, expandsTo.c_str());
         entries.push_back(toAdd);
       }
+
+      abortCounter++;
+      if (abortCounter > abortLimit)
+      {
+        ERR("Failure to parse and update Save File. Aborting update and loading clean slate.");
+        entries = {};
+        return;
+      }
     }
     in.close();