check length ensure msg is good for unpack

nlpng · Oct 10, 2024 · 9fc089b · 9fc089b
1 parent f9e5602
commit 9fc089b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -131,6 +131,12 @@ def main():
 
         data = msg[0]
         timestamp = msg[1]
+
+        if len(data) < 6:
+            # somehow data that is not enough to unpack
+            # the class, id, and payload length
+            continue
+
         if data[0] == SYNC1 and data[1] == SYNC2:
             try:
                 (cls, id, payload_len) = struct.unpack("<BBH", bytes(data[2:6]))

diff --git a/main_gui.py b/main_gui.py
@@ -81,6 +81,12 @@ def parsing(self):
 
             data = msg[0]
             timestamp = msg[1]
+
+            if len(data) < 6:
+                # somehow data that is not enough to unpack
+                # the class, id, and payload length
+                continue
+
             if data[0] == SYNC1 and data[1] == SYNC2:
                 try:
                     (cls, id, payload_len) = struct.unpack("<BBH", bytes(data[2:6]))