Handle decoding errors when parsing the character name

Wrap the UnicodeDecodeError exception and throw D2SFileParseError instead. Fixes #1
artcom-net · Oct 9, 2020 · e2cc4ea · e2cc4ea
1 parent 898e16d
commit e2cc4ea
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/d2lib/files.py b/d2lib/files.py
@@ -428,7 +428,13 @@ def _read_header(self):
             )
 
         self.active_weapon_slot = int_from_lbytes(self._reader.read(4))
-        self.char_name = self._reader.read(16).rstrip(b'\x00').decode('ASCII')
+
+        try:
+            self.char_name = (
+                self._reader.read(16).rstrip(b'\x00').decode('ASCII')
+            )
+        except UnicodeDecodeError as error:
+            raise D2SFileParseError(f'Error parsing character name: {error}')
 
         self.char_status = CharacterStatus(
             int_from_lbytes(self._reader.read(1))