|
@ -2,12 +2,12 @@ from __future__ import unicode_literals |
|
|
|
|
|
|
|
|
import collections |
|
|
import collections |
|
|
import io |
|
|
import io |
|
|
import struct |
|
|
|
|
|
import zlib |
|
|
import zlib |
|
|
|
|
|
|
|
|
from .utils import ( |
|
|
from .utils import ( |
|
|
compat_str, |
|
|
compat_str, |
|
|
ExtractorError, |
|
|
ExtractorError, |
|
|
|
|
|
struct_unpack, |
|
|
) |
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -23,17 +23,17 @@ def _extract_tags(file_contents): |
|
|
file_contents[:1]) |
|
|
file_contents[:1]) |
|
|
|
|
|
|
|
|
# Determine number of bits in framesize rectangle |
|
|
# Determine number of bits in framesize rectangle |
|
|
framesize_nbits = struct.unpack('!B', content[:1])[0] >> 3 |
|
|
|
|
|
|
|
|
framesize_nbits = struct_unpack('!B', content[:1])[0] >> 3 |
|
|
framesize_len = (5 + 4 * framesize_nbits + 7) // 8 |
|
|
framesize_len = (5 + 4 * framesize_nbits + 7) // 8 |
|
|
|
|
|
|
|
|
pos = framesize_len + 2 + 2 |
|
|
pos = framesize_len + 2 + 2 |
|
|
while pos < len(content): |
|
|
while pos < len(content): |
|
|
header16 = struct.unpack('<H', content[pos:pos + 2])[0] |
|
|
|
|
|
|
|
|
header16 = struct_unpack('<H', content[pos:pos + 2])[0] |
|
|
pos += 2 |
|
|
pos += 2 |
|
|
tag_code = header16 >> 6 |
|
|
tag_code = header16 >> 6 |
|
|
tag_len = header16 & 0x3f |
|
|
tag_len = header16 & 0x3f |
|
|
if tag_len == 0x3f: |
|
|
if tag_len == 0x3f: |
|
|
tag_len = struct.unpack('<I', content[pos:pos + 4])[0] |
|
|
|
|
|
|
|
|
tag_len = struct_unpack('<I', content[pos:pos + 4])[0] |
|
|
pos += 4 |
|
|
pos += 4 |
|
|
assert pos + tag_len <= len(content), \ |
|
|
assert pos + tag_len <= len(content), \ |
|
|
('Tag %d ends at %d+%d - that\'s longer than the file (%d)' |
|
|
('Tag %d ends at %d+%d - that\'s longer than the file (%d)' |
|
@ -99,7 +99,7 @@ def _read_int(reader): |
|
|
for _ in range(5): |
|
|
for _ in range(5): |
|
|
buf = reader.read(1) |
|
|
buf = reader.read(1) |
|
|
assert len(buf) == 1 |
|
|
assert len(buf) == 1 |
|
|
b = struct.unpack('<B', buf)[0] |
|
|
|
|
|
|
|
|
b = struct_unpack('<B', buf)[0] |
|
|
res = res | ((b & 0x7f) << shift) |
|
|
res = res | ((b & 0x7f) << shift) |
|
|
if b & 0x80 == 0: |
|
|
if b & 0x80 == 0: |
|
|
break |
|
|
break |
|
@ -125,7 +125,7 @@ def _s24(reader): |
|
|
bs = reader.read(3) |
|
|
bs = reader.read(3) |
|
|
assert len(bs) == 3 |
|
|
assert len(bs) == 3 |
|
|
last_byte = b'\xff' if (ord(bs[2:3]) >= 0x80) else b'\x00' |
|
|
last_byte = b'\xff' if (ord(bs[2:3]) >= 0x80) else b'\x00' |
|
|
return struct.unpack('<i', bs + last_byte)[0] |
|
|
|
|
|
|
|
|
return struct_unpack('<i', bs + last_byte)[0] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _read_string(reader): |
|
|
def _read_string(reader): |
|
@ -144,7 +144,7 @@ def _read_bytes(count, reader): |
|
|
|
|
|
|
|
|
def _read_byte(reader): |
|
|
def _read_byte(reader): |
|
|
resb = _read_bytes(1, reader=reader) |
|
|
resb = _read_bytes(1, reader=reader) |
|
|
res = struct.unpack('<B', resb)[0] |
|
|
|
|
|
|
|
|
res = struct_unpack('<B', resb)[0] |
|
|
return res |
|
|
return res |
|
|
|
|
|
|
|
|
|
|
|
|
|
|