mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-04 18:33:38 +00:00
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
"""
|
|
Apple BOMStorage parser.
|
|
|
|
Used for Assets.Bom files by Interface Builder, and for .bom files by Installer.app.
|
|
|
|
Documents:
|
|
|
|
Author: Robert Xiao
|
|
Created: 2015-05-14
|
|
"""
|
|
|
|
from hachoir_parser import HachoirParser
|
|
from hachoir_core.field import (RootSeekableFieldSet, FieldSet, Enum,
|
|
Bits, GenericInteger, Float32, Float64, UInt8, UInt32, UInt64, Bytes, NullBytes, RawBytes, String)
|
|
from hachoir_core.endian import BIG_ENDIAN
|
|
from hachoir_core.text_handler import displayHandler
|
|
from hachoir_core.tools import humanDatetime
|
|
from datetime import datetime, timedelta
|
|
|
|
class BomTrailerEntry(FieldSet):
|
|
static_size = 64 # bits
|
|
def createFields(self):
|
|
yield UInt32(self, "offset")
|
|
yield UInt32(self, "size")
|
|
def createDescription(self):
|
|
return "Object at offset %d, size %d" % (self['offset'].value, self['size'].value)
|
|
|
|
class BomTrailer(FieldSet):
|
|
def createFields(self):
|
|
yield UInt32(self, "num_spaces", "Total number of entries, including blank entries")
|
|
nobj = self['/num_objects'].value
|
|
nspace = self['num_spaces'].value
|
|
for i in xrange(nobj+1):
|
|
yield BomTrailerEntry(self, "entry[]")
|
|
yield NullBytes(self, "blank_entries", (nspace - nobj - 1) * (BomTrailerEntry.static_size / 8))
|
|
yield UInt32(self, "num_trail")
|
|
ntrail = self['num_trail'].value
|
|
for i in xrange(ntrail):
|
|
yield BomTrailerEntry(self, "trail[]")
|
|
|
|
def createDescription(self):
|
|
return "Bom file trailer"
|
|
|
|
class BomFile(HachoirParser, RootSeekableFieldSet):
|
|
endian = BIG_ENDIAN
|
|
MAGIC = "BOMStore"
|
|
PARSER_TAGS = {
|
|
"id": "bom_store",
|
|
"category": "archive",
|
|
"file_ext": ("bom","car"),
|
|
"magic": ((MAGIC, 0),),
|
|
"min_size": 32, # 32-byte header
|
|
"description": "Apple bill-of-materials file",
|
|
}
|
|
|
|
def __init__(self, stream, **args):
|
|
RootSeekableFieldSet.__init__(self, None, "root", stream, None, stream.askSize(self))
|
|
HachoirParser.__init__(self, stream, **args)
|
|
|
|
def validate(self):
|
|
if self.stream.readBytes(0, len(self.MAGIC)) != self.MAGIC:
|
|
return "Invalid magic"
|
|
return True
|
|
|
|
def createFields(self):
|
|
yield Bytes(self, "magic", 8, "File magic (BOMStore)")
|
|
yield UInt32(self, "version") # ?
|
|
yield UInt32(self, "num_objects")
|
|
yield UInt32(self, "trailer_offset")
|
|
yield UInt32(self, "trailer_size")
|
|
yield UInt32(self, "header_offset")
|
|
yield UInt32(self, "header_size")
|
|
|
|
yield RawBytes(self, "object[]", 512-32, "Null object (size 0, offset 0)") # null object
|
|
|
|
self.seekByte(self['trailer_offset'].value)
|
|
yield BomTrailer(self, "trailer")
|
|
|
|
self.seekByte(self['header_offset'].value)
|
|
yield RawBytes(self, "header", self['header_size'].value)
|
|
|
|
for entry in self['trailer'].array('entry'):
|
|
if entry['size'].value == 0:
|
|
continue
|
|
self.seekByte(entry['offset'].value)
|
|
yield RawBytes(self, "object[]", entry['size'].value)
|
|
|
|
for entry in self['trailer'].array('trail'):
|
|
self.seekByte(entry['offset'].value)
|
|
yield RawBytes(self, "trail[]", entry['size'].value)
|