Resolve Source Engine Protocol issue

Author: BattlefieldDuck

opengsq/protocols/source.py

@@ -101,24 +101,26 @@ def __parse_from_info_src(self, br: BinaryReader) -> SourceInfo:
             info["duration"] = br.read_byte()
 
         info["version"] = br.read_string()
-        edf = ExtraDataFlag(br.read_byte())
-        info["edf"] = edf
 
-        if edf.has_flag(ExtraDataFlag.Port):
-            info["port"] = br.read_short()
+        if not br.is_end():
+            edf = ExtraDataFlag(br.read_byte())
+            info["edf"] = edf
 
-        if edf.has_flag(ExtraDataFlag.SteamID):
-            info["steam_id"] = br.read_long_long()
+            if edf.has_flag(ExtraDataFlag.Port):
+                info["port"] = br.read_short()
 
-        if edf.has_flag(ExtraDataFlag.Spectator):
-            info["spectator_port"] = br.read_short()
-            info["spectator_name"] = br.read_string()
+            if edf.has_flag(ExtraDataFlag.SteamID):
+                info["steam_id"] = br.read_long_long()
 
-        if edf.has_flag(ExtraDataFlag.Keywords):
-            info["keywords"] = br.read_string()
+            if edf.has_flag(ExtraDataFlag.Spectator):
+                info["spectator_port"] = br.read_short()
+                info["spectator_name"] = br.read_string()
 
-        if edf.has_flag(ExtraDataFlag.GameID):
-            info["game_id"] = br.read_long_long()
+            if edf.has_flag(ExtraDataFlag.Keywords):
+                info["keywords"] = br.read_string()
+
+            if edf.has_flag(ExtraDataFlag.GameID):
+                info["game_id"] = br.read_long_long()
 
         return SourceInfo(**info)
 

opengsq/responses/source/source_info.py

@@ -16,7 +16,7 @@ class SourceInfo(PartialInfo):
     version: str
     """Version of the game installed on the server."""
 
-    edf: ExtraDataFlag
+    edf: Optional[ExtraDataFlag] = None
     """If present, this specifies which additional data fields will be included."""
 
     port: Optional[int] = None

opengsq/version.py

@@ -1 +1 @@
-__version__ = '3.0.8'
+__version__ = '3.0.9'