add a range check to EditorUI

geode-sdk · May 22, 2024 · 73dd32b · 73dd32b
1 parent ae92005
commit 73dd32b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/EditorUI.cpp b/src/EditorUI.cpp
@@ -240,6 +240,7 @@ using namespace geode::node_ids;
         if (auto bar = getChildOfType<EditButtonBar>(this, buttonBarIndex++)) {
             bar->setID(id);
         }
+        else break;
     }
 
     if (auto editBar = static_cast<EditButtonBar*>(this->getChildByID("edit-tab-bar"))) {
@@ -278,7 +279,9 @@ using namespace geode::node_ids;
         };
         size_t i = 0;
         for (auto id : editButtonIDs) {
-            static_cast<CCNode*>(editBar->m_buttonArray->objectAtIndex(i++))->setID(id);
+            if (i < editBar->m_buttonArray->count()) {
+                static_cast<CCNode*>(editBar->m_buttonArray->objectAtIndex(i++))->setID(id);
+            }
         }
     }