1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
##=============================================================================
#
# Copyright (C) 2003, 2011 Alessandro Duca <alessandro.duca@gmail.com>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
#=============================================================================
#
##=============================================================================
import yenc
import os
import sys
import re
NAME_RE = re.compile(br"^.*? name=(.+?)\r\n$")
LINE_RE = re.compile(br"^.*? line=(\d{3}) .*$")
SIZE_RE = re.compile(br"^.*? size=(\d+) .*$")
CRC32_RE = re.compile(br"^.*? crc32=(\w+)")
def main():
head_crc = trail_crc = None
if len(sys.argv) > 1:
file_in = open(sys.argv[1],"rb")
else:
try: # Python 3
file_in = sys.stdin.detach()
except AttributeError: # Python < 3
file_in = sys.stdin
sys.stdin = None
with file_in:
while True:
line = file_in.readline()
if line.startswith(b"=ybegin "):
try:
name, size = NAME_RE.match(line).group(1), int(SIZE_RE.match(line).group(1))
m_obj = CRC32_RE.match(line)
if m_obj:
head_crc = m_obj.group(1)
except re.error:
sys.stderr.write("err-critical: malformed =ybegin header\n")
sys.exit(1)
break
elif not line:
sys.stderr.write("err-critical: no valid =ybegin header found\n")
sys.exit(1)
with open(name,"wb") as file_out:
try:
dec, dec_crc = yenc.decode(file_in, file_out, size)
except yenc.Error as e:
sys.stderr.write(str(e) + '\n')
sys.exit(1)
garbage = False
for line in file_in.read().split(b"\r\n"):
if line.startswith(b"=yend "):
try:
size = int( SIZE_RE.match(line).group(1) )
m_obj = CRC32_RE.match(line)
if m_obj:
trail_crc = m_obj.group(1)
except re.error:
sys.stderr.write("err: malformed =yend trailer\n")
break
elif not line:
continue
else:
garbage = True
else:
sys.stderr.write("warning: couldn't find =yend trailer\n")
if garbage:
sys.stderr.write("warning: garbage before =yend trailer\n")
if head_crc:
tmp_crc = head_crc.decode("ascii").lower()
elif trail_crc:
tmp_crc = trail_crc.decode("ascii").lower()
else:
sys.exit(0)
if tmp_crc != dec_crc:
sys.stderr.write("err: header: %s dec: %s CRC32 mismatch\n" % (tmp_crc,dec_crc) )
sys.exit(1)
else:
sys.exit(0)
if __name__ == "__main__":
main()
|
