You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
214 lines
7.1 KiB
214 lines
7.1 KiB
#! /usr/bin/env python3
|
|
|
|
# Copyright 1994 by Lance Ellinghouse
|
|
# Cathedral City, California Republic, United States of America.
|
|
# All Rights Reserved
|
|
# Permission to use, copy, modify, and distribute this software and its
|
|
# documentation for any purpose and without fee is hereby granted,
|
|
# provided that the above copyright notice appear in all copies and that
|
|
# both that copyright notice and this permission notice appear in
|
|
# supporting documentation, and that the name of Lance Ellinghouse
|
|
# not be used in advertising or publicity pertaining to distribution
|
|
# of the software without specific, written prior permission.
|
|
# LANCE ELLINGHOUSE DISCLAIMS ALL WARRANTIES WITH REGARD TO
|
|
# THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
|
|
# FITNESS, IN NO EVENT SHALL LANCE ELLINGHOUSE CENTRUM BE LIABLE
|
|
# FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
|
|
# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
#
|
|
# Modified by Jack Jansen, CWI, July 1995:
|
|
# - Use binascii module to do the actual line-by-line conversion
|
|
# between ascii and binary. This results in a 1000-fold speedup. The C
|
|
# version is still 5 times faster, though.
|
|
# - Arguments more compliant with python standard
|
|
|
|
"""Implementation of the UUencode and UUdecode functions.
|
|
|
|
encode(in_file, out_file [,name, mode], *, backtick=False)
|
|
decode(in_file [, out_file, mode, quiet])
|
|
"""
|
|
|
|
import binascii
|
|
import os
|
|
import sys
|
|
|
|
__all__ = ["Error", "encode", "decode"]
|
|
|
|
class Error(Exception):
|
|
pass
|
|
|
|
def encode(in_file, out_file, name=None, mode=None, *, backtick=False):
|
|
"""Uuencode file"""
|
|
#
|
|
# If in_file is a pathname open it and change defaults
|
|
#
|
|
opened_files = []
|
|
try:
|
|
if in_file == '-':
|
|
in_file = sys.stdin.buffer
|
|
elif isinstance(in_file, str):
|
|
if name is None:
|
|
name = os.path.basename(in_file)
|
|
if mode is None:
|
|
try:
|
|
mode = os.stat(in_file).st_mode
|
|
except AttributeError:
|
|
pass
|
|
in_file = open(in_file, 'rb')
|
|
opened_files.append(in_file)
|
|
#
|
|
# Open out_file if it is a pathname
|
|
#
|
|
if out_file == '-':
|
|
out_file = sys.stdout.buffer
|
|
elif isinstance(out_file, str):
|
|
out_file = open(out_file, 'wb')
|
|
opened_files.append(out_file)
|
|
#
|
|
# Set defaults for name and mode
|
|
#
|
|
if name is None:
|
|
name = '-'
|
|
if mode is None:
|
|
mode = 0o666
|
|
|
|
#
|
|
# Remove newline chars from name
|
|
#
|
|
name = name.replace('\n','\\n')
|
|
name = name.replace('\r','\\r')
|
|
|
|
#
|
|
# Write the data
|
|
#
|
|
out_file.write(('begin %o %s\n' % ((mode & 0o777), name)).encode("ascii"))
|
|
data = in_file.read(45)
|
|
while len(data) > 0:
|
|
out_file.write(binascii.b2a_uu(data, backtick=backtick))
|
|
data = in_file.read(45)
|
|
if backtick:
|
|
out_file.write(b'`\nend\n')
|
|
else:
|
|
out_file.write(b' \nend\n')
|
|
finally:
|
|
for f in opened_files:
|
|
f.close()
|
|
|
|
|
|
def decode(in_file, out_file=None, mode=None, quiet=False):
|
|
"""Decode uuencoded file"""
|
|
#
|
|
# Open the input file, if needed.
|
|
#
|
|
opened_files = []
|
|
if in_file == '-':
|
|
in_file = sys.stdin.buffer
|
|
elif isinstance(in_file, str):
|
|
in_file = open(in_file, 'rb')
|
|
opened_files.append(in_file)
|
|
|
|
try:
|
|
#
|
|
# Read until a begin is encountered or we've exhausted the file
|
|
#
|
|
while True:
|
|
hdr = in_file.readline()
|
|
if not hdr:
|
|
raise Error('No valid begin line found in input file')
|
|
if not hdr.startswith(b'begin'):
|
|
continue
|
|
hdrfields = hdr.split(b' ', 2)
|
|
if len(hdrfields) == 3 and hdrfields[0] == b'begin':
|
|
try:
|
|
int(hdrfields[1], 8)
|
|
break
|
|
except ValueError:
|
|
pass
|
|
if out_file is None:
|
|
# If the filename isn't ASCII, what's up with that?!?
|
|
out_file = hdrfields[2].rstrip(b' \t\r\n\f').decode("ascii")
|
|
if os.path.exists(out_file):
|
|
raise Error(f'Cannot overwrite existing file: {out_file}')
|
|
if (out_file.startswith(os.sep) or
|
|
f'..{os.sep}' in out_file or (
|
|
os.altsep and
|
|
(out_file.startswith(os.altsep) or
|
|
f'..{os.altsep}' in out_file))
|
|
):
|
|
raise Error(f'Refusing to write to {out_file} due to directory traversal')
|
|
if mode is None:
|
|
mode = int(hdrfields[1], 8)
|
|
#
|
|
# Open the output file
|
|
#
|
|
if out_file == '-':
|
|
out_file = sys.stdout.buffer
|
|
elif isinstance(out_file, str):
|
|
fp = open(out_file, 'wb')
|
|
os.chmod(out_file, mode)
|
|
out_file = fp
|
|
opened_files.append(out_file)
|
|
#
|
|
# Main decoding loop
|
|
#
|
|
s = in_file.readline()
|
|
while s and s.strip(b' \t\r\n\f') != b'end':
|
|
try:
|
|
data = binascii.a2b_uu(s)
|
|
except binascii.Error as v:
|
|
# Workaround for broken uuencoders by /Fredrik Lundh
|
|
nbytes = (((s[0]-32) & 63) * 4 + 5) // 3
|
|
data = binascii.a2b_uu(s[:nbytes])
|
|
if not quiet:
|
|
sys.stderr.write("Warning: %s\n" % v)
|
|
out_file.write(data)
|
|
s = in_file.readline()
|
|
if not s:
|
|
raise Error('Truncated input file')
|
|
finally:
|
|
for f in opened_files:
|
|
f.close()
|
|
|
|
def test():
|
|
"""uuencode/uudecode main program"""
|
|
|
|
import optparse
|
|
parser = optparse.OptionParser(usage='usage: %prog [-d] [-t] [input [output]]')
|
|
parser.add_option('-d', '--decode', dest='decode', help='Decode (instead of encode)?', default=False, action='store_true')
|
|
parser.add_option('-t', '--text', dest='text', help='data is text, encoded format unix-compatible text?', default=False, action='store_true')
|
|
|
|
(options, args) = parser.parse_args()
|
|
if len(args) > 2:
|
|
parser.error('incorrect number of arguments')
|
|
sys.exit(1)
|
|
|
|
# Use the binary streams underlying stdin/stdout
|
|
input = sys.stdin.buffer
|
|
output = sys.stdout.buffer
|
|
if len(args) > 0:
|
|
input = args[0]
|
|
if len(args) > 1:
|
|
output = args[1]
|
|
|
|
if options.decode:
|
|
if options.text:
|
|
if isinstance(output, str):
|
|
output = open(output, 'wb')
|
|
else:
|
|
print(sys.argv[0], ': cannot do -t to stdout')
|
|
sys.exit(1)
|
|
decode(input, output)
|
|
else:
|
|
if options.text:
|
|
if isinstance(input, str):
|
|
input = open(input, 'rb')
|
|
else:
|
|
print(sys.argv[0], ': cannot do -t from stdin')
|
|
sys.exit(1)
|
|
encode(input, output)
|
|
|
|
if __name__ == '__main__':
|
|
test()
|