ResourcePack: Avoid crashes on invalid packs during Init().

2 files changed, 29 insertions, 20 deletions

diff --git a/Source/Core/UICommon/ResourcePack/Manager.cpp b/Source/Core/UICommon/ResourcePack/Manager.cpp
index f83a15dd8a..2d1cf128ec 100644
--- a/Source/Core/UICommon/ResourcePack/Manager.cpp
+++ b/Source/Core/UICommon/ResourcePack/Manager.cpp
@@ -31,35 +31,45 @@ IniFile GetPackConfig()
 bool Init()
 {
   packs.clear();
-  auto pack_list = Common::DoFileSearch({File::GetUserPath(D_RESOURCEPACK_IDX)}, {".zip"});
-
-  bool error = false;
+  const std::vector<std::string> pack_list =
+      Common::DoFileSearch({File::GetUserPath(D_RESOURCEPACK_IDX)}, {".zip"});
 
   IniFile file = GetPackConfig();
 
   auto* order = file.GetOrCreateSection("Order");
 
-  std::sort(pack_list.begin(), pack_list.end(), [order](std::string& a, std::string& b) {
-    std::string order_a = a, order_b = b;
+  struct OrderHelper
+  {
+    size_t pack_list_index;
+    std::string manifest_id;
+  };
 
-    order->Get(ResourcePack(a).GetManifest()->GetID(), &order_a);
-    order->Get(ResourcePack(b).GetManifest()->GetID(), &order_b);
+  std::vector<OrderHelper> pack_list_order;
+  pack_list_order.reserve(pack_list.size());
+  for (size_t i = 0; i < pack_list.size(); ++i)
+  {
+    const ResourcePack pack(pack_list[i]);
+    std::string manifest_id = pack.IsValid() ? pack.GetManifest()->GetID() : pack_list[i];
+    pack_list_order.emplace_back(OrderHelper{i, std::move(manifest_id)});
+  }
 
-    return order_a < order_b;
-  });
+  std::sort(
+      pack_list_order.begin(), pack_list_order.end(),
+      [](const OrderHelper& a, const OrderHelper& b) { return a.manifest_id < b.manifest_id; });
 
-  for (size_t i = 0; i < pack_list.size(); i++)
+  bool error = false;
+  for (size_t i = 0; i < pack_list_order.size(); ++i)
   {
-    const auto& path = pack_list[i];
+    const auto& path = pack_list[pack_list_order[i].pack_list_index];
 
-    if (!Add(path))
+    const ResourcePack* const pack = Add(path);
+    if (pack == nullptr)
     {
       error = true;
       continue;
     }
 
-    if (i < packs.size())
-      order->Set(packs[i].GetManifest()->GetID(), static_cast<u64>(i));
+    order->Set(pack->GetManifest()->GetID(), static_cast<u64>(i));
   }
 
   file.Save(packs_path);
@@ -103,7 +113,7 @@ std::vector<ResourcePack*> GetHigherPriorityPacks(ResourcePack& pack)
   return list;
 }
 
-bool Add(const std::string& path, int offset)
+ResourcePack* Add(const std::string& path, int offset)
 {
   if (offset == -1)
     offset = static_cast<int>(packs.size());
@@ -111,7 +121,7 @@ bool Add(const std::string& path, int offset)
   ResourcePack pack(path);
 
   if (!pack.IsValid())
-    return false;
+    return nullptr;
 
   IniFile file = GetPackConfig();
 
@@ -124,9 +134,8 @@ bool Add(const std::string& path, int offset)
 
   file.Save(packs_path);
 
-  packs.insert(packs.begin() + offset, std::move(pack));
-
-  return true;
+  auto it = packs.insert(packs.begin() + offset, std::move(pack));
+  return &*it;
 }
 
 bool Remove(ResourcePack& pack)
diff --git a/Source/Core/UICommon/ResourcePack/Manager.h b/Source/Core/UICommon/ResourcePack/Manager.h
index e31e780b52..da81c7d40b 100644
--- a/Source/Core/UICommon/ResourcePack/Manager.h
+++ b/Source/Core/UICommon/ResourcePack/Manager.h
@@ -12,7 +12,7 @@ namespace ResourcePack
 {
 bool Init();
 
-bool Add(const std::string& path, int offset = -1);
+ResourcePack* Add(const std::string& path, int offset = -1);
 bool Remove(ResourcePack& pack);
 void SetInstalled(const ResourcePack& pack, bool installed);
 bool IsInstalled(const ResourcePack& pack);