mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-16 18:05:04 +00:00
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
|
"""
|
||
|
GNU ar archive : archive file (.a) and Debian (.deb) archive.
|
||
|
"""
|
||
|
|
||
|
from hachoir.parser import Parser
|
||
|
from hachoir.field import (FieldSet, ParserError,
|
||
|
String, RawBytes, UnixLine)
|
||
|
from hachoir.core.endian import BIG_ENDIAN
|
||
|
|
||
|
|
||
|
class ArchiveFileEntry(FieldSet):
|
||
|
|
||
|
def createFields(self):
|
||
|
yield UnixLine(self, "header", "Header")
|
||
|
info = self["header"].value
|
||
|
info = info.split()
|
||
|
if len(info) < 3:
|
||
|
raise ParserError("Invalid file entry header: %r" % info)
|
||
|
size = int(info[-2])
|
||
|
if 0 < size:
|
||
|
yield RawBytes(self, "content", size, "File data")
|
||
|
|
||
|
def createDescription(self):
|
||
|
return "File entry (%s)" % self["header"].value.split()[0]
|
||
|
|
||
|
|
||
|
class ArchiveFile(Parser):
|
||
|
endian = BIG_ENDIAN
|
||
|
MAGIC = b'!<arch>\n'
|
||
|
PARSER_TAGS = {
|
||
|
"id": "unix_archive",
|
||
|
"category": "archive",
|
||
|
"file_ext": ("a", "deb"),
|
||
|
"mime":
|
||
|
("application/x-debian-package",
|
||
|
"application/x-archive",
|
||
|
"application/x-dpkg"),
|
||
|
"min_size": (8 + 13) * 8, # file signature + smallest file as possible
|
||
|
"magic": ((MAGIC, 0),),
|
||
|
"description": "Unix archive"
|
||
|
}
|
||
|
|
||
|
def validate(self):
|
||
|
if self.stream.readBytes(0, len(self.MAGIC)) != self.MAGIC:
|
||
|
return "Invalid magic string"
|
||
|
return True
|
||
|
|
||
|
def createFields(self):
|
||
|
yield String(self, "id", 8, "Unix archive identifier (\"<!arch>\")", charset="ASCII")
|
||
|
while not self.eof:
|
||
|
data = self.stream.readBytes(self.current_size, 1)
|
||
|
if data == b"\n":
|
||
|
yield RawBytes(self, "empty_line[]", 1, "Empty line")
|
||
|
else:
|
||
|
yield ArchiveFileEntry(self, "file[]", "File")
|