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
|
from pypy.interpreter.gateway import unwrap_spec
from rpython.rlib.rstring import StringBuilder
from pypy.module.binascii.interp_binascii import raise_Error
from pypy.module.binascii.interp_binascii import AsciiBufferUnwrapper
# ____________________________________________________________
def _a2b_read(space, s, index):
try:
c = s[index]
except IndexError:
return 0
# Check the character for legality. The 64 instead of the expected 63
# is because there are a few uuencodes out there that use '`' as zero
# instead of space.
if c < ' ' or c > chr(32 + 64):
if c == '\n' or c == '\r':
return 0
raise_Error(space, "Illegal char")
return (ord(c) - 0x20) & 0x3f
_a2b_read._always_inline_ = True
@unwrap_spec(ascii=AsciiBufferUnwrapper)
def a2b_uu(space, ascii):
"Decode a line of uuencoded data."
if len(ascii) == 0: # obscure case, for compability with CPython
length = (-0x20) & 0x3f
else:
length = (ord(ascii[0]) - 0x20) & 0x3f
res = StringBuilder(length)
for i in range(1, len(ascii), 4):
A = _a2b_read(space, ascii, i)
B = _a2b_read(space, ascii, i+1)
C = _a2b_read(space, ascii, i+2)
D = _a2b_read(space, ascii, i+3)
#
if res.getlength() < length:
res.append(chr(A << 2 | B >> 4))
elif A != 0 or B != 0:
raise_Error(space, "Trailing garbage")
#
if res.getlength() < length:
res.append(chr((B & 0xf) << 4 | C >> 2))
elif C != 0:
raise_Error(space, "Trailing garbage")
#
if res.getlength() < length:
res.append(chr((C & 0x3) << 6 | D))
elif D != 0:
raise_Error(space, "Trailing garbage")
remaining = length - res.getlength()
if remaining > 0:
res.append_multiple_char('\x00', remaining)
return space.newbytes(res.build())
# ____________________________________________________________
def _b2a_read(bin, i):
try:
return ord(bin[i])
except IndexError:
return 0
_b2a_read._always_inline_ = True
def _b2a_write(res, num, backtick):
if backtick and not num:
res.append(chr(0x60))
else:
res.append(chr(0x20 + num))
_b2a_write._always_inline_ = True
@unwrap_spec(bin='bufferstr', backtick=bool)
def b2a_uu(space, bin, __kwonly__, backtick=False):
"Uuencode a line of data."
length = len(bin)
if length > 45:
raise_Error(space, 'At most 45 bytes at once')
res = StringBuilder(2 + ((length + 2) // 3) * 4)
_b2a_write(res, length, backtick)
for i in range(0, length, 3):
A = _b2a_read(bin, i)
B = _b2a_read(bin, i+1)
C = _b2a_read(bin, i+2)
#
_b2a_write(res, A >> 2, backtick)
_b2a_write(res, (A & 0x3) << 4 | B >> 4, backtick)
_b2a_write(res, (B & 0xF) << 2 | C >> 6, backtick)
_b2a_write(res, C & 0x3F , backtick)
res.append('\n')
return space.newbytes(res.build())
|
