mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-22 12:55:05 +00:00
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
|
from hachoir.core.endian import BIG_ENDIAN, LITTLE_ENDIAN
|
||
|
|
from hachoir.field import RawBytes, RootSeekableFieldSet, ParserError
|
||
|
|
from hachoir.parser import HachoirParser
|
||
|
|
|
||
|
|
|
||
|
|
class OLE2FragmentParser(HachoirParser, RootSeekableFieldSet):
|
||
|
|
tags = {
|
||
|
|
"description": "Microsoft Office document subfragments",
|
||
|
|
}
|
||
|
|
endian = LITTLE_ENDIAN
|
||
|
|
|
||
|
|
ENDIAN_CHECK = False
|
||
|
|
|
||
|
|
def __init__(self, stream, **args):
|
||
|
|
RootSeekableFieldSet.__init__(
|
||
|
|
self, None, "root", stream, None, stream.askSize(self))
|
||
|
|
HachoirParser.__init__(self, stream, **args)
|
||
|
|
if self.ENDIAN_CHECK:
|
||
|
|
if self["endian"].value == b"\xFF\xFE":
|
||
|
|
self.endian = BIG_ENDIAN
|
||
|
|
elif self["endian"].value == b"\xFE\xFF":
|
||
|
|
self.endian = LITTLE_ENDIAN
|
||
|
|
else:
|
||
|
|
raise ParserError("OLE2: Invalid endian value")
|
||
|
|
|
||
|
|
def validate(self):
|
||
|
|
if self.ENDIAN_CHECK:
|
||
|
|
if self["endian"].value not in ["\xFF\xFE", "\xFE\xFF"]:
|
||
|
|
return "Unknown endian value %s" % self["endian"].value.encode('hex')
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
class RawParser(OLE2FragmentParser):
|
||
|
|
ENDIAN_CHECK = False
|
||
|
|
OS_CHECK = False
|
||
|
|
|
||
|
|
def createFields(self):
|
||
|
|
yield RawBytes(self, "rawdata", self.datasize)
|
||
|
|
if self.datasize < self.size // 8:
|
||
|
|
yield RawBytes(self, "slack_space", (self.size // 8) - self.datasize)
|