Leahnaya/kbinxml
1
0
Fork 0
mirror of https://github.com/mon/kbinxml.git synced 2026-04-26 02:01:45 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

is_binary_xml: don't explode on too-short inputs

Browse Source
This commit is contained in:
William Toohey 2021-05-03 22:32:27 +10:00
parent 3cbc3179a8
commit 5707a6bf8b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
kbinxml/kbinxml.py
View File

@ -68,6 +68,9 @@ class KBinXML():
@staticmethod @staticmethod
def is_binary_xml(input): def is_binary_xml(input):
if len(input) < 2:
return False
nodeBuf = ByteBuffer(input) nodeBuf = ByteBuffer(input)
return (nodeBuf.get_u8() == SIGNATURE and return (nodeBuf.get_u8() == SIGNATURE and
nodeBuf.get_u8() in (SIG_COMPRESSED, SIG_UNCOMPRESSED)) nodeBuf.get_u8() in (SIG_COMPRESSED, SIG_UNCOMPRESSED))