Fix bug in Ubi identifier

redballoonsecurity · Dec 14, 2024 · 47c1486 · 47c1486
1 parent 6d053dd
commit 47c1486
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ofrak_core/ofrak/core/ubi.py b/ofrak_core/ofrak/core/ubi.py
@@ -304,7 +304,7 @@ async def pack(self, resource: Resource, config=None) -> None:
 def match_ubi_magic(data: bytes) -> bool:
     if len(data) < 4:
         return False
-    return data in [UBI_EC_HDR_MAGIC, UBI_VID_HDR_MAGIC]
+    return data[:4] in [UBI_EC_HDR_MAGIC, UBI_VID_HDR_MAGIC]
 
 
 RawMagicIdentifier.register(Ubi, match_ubi_magic)