Fixed an crash when short seed numbers are used

mcgrew · Jun 7, 2016 · 81c8e5e · 81c8e5e
1 parent a6836cf
commit 81c8e5e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dwrandomizer.py b/dwrandomizer.py
@@ -685,6 +685,10 @@ def update_title_screen(self, seed, flags):
     new_text += blank_line
 
     needed_bytes = len(self.title_screen_text) - len(new_text) - 4
+    if needed_bytes > 31: # current data is too short...
+      new_text = new_text[:-len(blank_line)] + (b'\x5f' * 32 + b'\xfc')
+      needed_bytes = len(self.title_screen_text) - len(new_text) - 4
+
     new_text += b'\x5f' * needed_bytes + padding(32 - needed_bytes) + b'\xfc'
     new_text = new_text.replace(b'\x47', b'\x61').replace(b'\x49', b'\x63')