1 #! /usr/bin/env python3 2 3 # Copyright 1994 by Lance Ellinghouse 4 # Cathedral City, California Republic, United States of America. 5 # All Rights Reserved 6 # Permission to use, copy, modify, and distribute this software and its 7 # documentation for any purpose and without fee is hereby granted, 8 # provided that the above copyright notice appear in all copies and that 9 # both that copyright notice and this permission notice appear in 10 # supporting documentation, and that the name of Lance Ellinghouse 11 # not be used in advertising or publicity pertaining to distribution 12 # of the software without specific, written prior permission. 13 # LANCE ELLINGHOUSE DISCLAIMS ALL WARRANTIES WITH REGARD TO 14 # THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND 15 # FITNESS, IN NO EVENT SHALL LANCE ELLINGHOUSE CENTRUM BE LIABLE 16 # FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 17 # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 18 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT 19 # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 20 # 21 # Modified by Jack Jansen, CWI, July 1995: 22 # - Use binascii module to do the actual line-by-line conversion 23 # between ascii and binary. This results in a 1000-fold speedup. The C 24 # version is still 5 times faster, though. 25 # - Arguments more compliant with python standard 26 27 """Implementation of the UUencode and UUdecode functions. 28 29 encode(in_file, out_file [,name, mode], *, backtick=False) 30 decode(in_file [, out_file, mode, quiet]) 31 """ 32 33 import binascii 34 import os 35 import sys 36 37 __all__ = ["Error", "encode", "decode"] 38 39 class Error(Exception): 40 pass 41 42 def encode(in_file, out_file, name=None, mode=None, *, backtick=False): 43 """Uuencode file""" 44 # 45 # If in_file is a pathname open it and change defaults 46 # 47 opened_files = [] 48 try: 49 if in_file == '-': 50 in_file = sys.stdin.buffer 51 elif isinstance(in_file, str): 52 if name is None: 53 name = os.path.basename(in_file) 54 if mode is None: 55 try: 56 mode = os.stat(in_file).st_mode 57 except AttributeError: 58 pass 59 in_file = open(in_file, 'rb') 60 opened_files.append(in_file) 61 # 62 # Open out_file if it is a pathname 63 # 64 if out_file == '-': 65 out_file = sys.stdout.buffer 66 elif isinstance(out_file, str): 67 out_file = open(out_file, 'wb') 68 opened_files.append(out_file) 69 # 70 # Set defaults for name and mode 71 # 72 if name is None: 73 name = '-' 74 if mode is None: 75 mode = 0o666 76 # 77 # Write the data 78 # 79 out_file.write(('begin %o %s\n' % ((mode & 0o777), name)).encode("ascii")) 80 data = in_file.read(45) 81 while len(data) > 0: 82 out_file.write(binascii.b2a_uu(data, backtick=backtick)) 83 data = in_file.read(45) 84 if backtick: 85 out_file.write(b'`\nend\n') 86 else: 87 out_file.write(b' \nend\n') 88 finally: 89 for f in opened_files: 90 f.close() 91 92 93 def decode(in_file, out_file=None, mode=None, quiet=False): 94 """Decode uuencoded file""" 95 # 96 # Open the input file, if needed. 97 # 98 opened_files = [] 99 if in_file == '-': 100 in_file = sys.stdin.buffer 101 elif isinstance(in_file, str): 102 in_file = open(in_file, 'rb') 103 opened_files.append(in_file) 104 105 try: 106 # 107 # Read until a begin is encountered or we've exhausted the file 108 # 109 while True: 110 hdr = in_file.readline() 111 if not hdr: 112 raise Error('No valid begin line found in input file') 113 if not hdr.startswith(b'begin'): 114 continue 115 hdrfields = hdr.split(b' ', 2) 116 if len(hdrfields) == 3 and hdrfields[0] == b'begin': 117 try: 118 int(hdrfields[1], 8) 119 break 120 except ValueError: 121 pass 122 if out_file is None: 123 # If the filename isn't ASCII, what's up with that?!? 124 out_file = hdrfields[2].rstrip(b' \t\r\n\f').decode("ascii") 125 if os.path.exists(out_file): 126 raise Error('Cannot overwrite existing file: %s' % out_file) 127 if mode is None: 128 mode = int(hdrfields[1], 8) 129 # 130 # Open the output file 131 # 132 if out_file == '-': 133 out_file = sys.stdout.buffer 134 elif isinstance(out_file, str): 135 fp = open(out_file, 'wb') 136 os.chmod(out_file, mode) 137 out_file = fp 138 opened_files.append(out_file) 139 # 140 # Main decoding loop 141 # 142 s = in_file.readline() 143 while s and s.strip(b' \t\r\n\f') != b'end': 144 try: 145 data = binascii.a2b_uu(s) 146 except binascii.Error as v: 147 # Workaround for broken uuencoders by /Fredrik Lundh 148 nbytes = (((s[0]-32) & 63) * 4 + 5) // 3 149 data = binascii.a2b_uu(s[:nbytes]) 150 if not quiet: 151 sys.stderr.write("Warning: %s\n" % v) 152 out_file.write(data) 153 s = in_file.readline() 154 if not s: 155 raise Error('Truncated input file') 156 finally: 157 for f in opened_files: 158 f.close() 159 160 def test(): 161 """uuencode/uudecode main program""" 162 163 import optparse 164 parser = optparse.OptionParser(usage='usage: %prog [-d] [-t] [input [output]]') 165 parser.add_option('-d', '--decode', dest='decode', help='Decode (instead of encode)?', default=False, action='store_true') 166 parser.add_option('-t', '--text', dest='text', help='data is text, encoded format unix-compatible text?', default=False, action='store_true') 167 168 (options, args) = parser.parse_args() 169 if len(args) > 2: 170 parser.error('incorrect number of arguments') 171 sys.exit(1) 172 173 # Use the binary streams underlying stdin/stdout 174 input = sys.stdin.buffer 175 output = sys.stdout.buffer 176 if len(args) > 0: 177 input = args[0] 178 if len(args) > 1: 179 output = args[1] 180 181 if options.decode: 182 if options.text: 183 if isinstance(output, str): 184 output = open(output, 'wb') 185 else: 186 print(sys.argv[0], ': cannot do -t to stdout') 187 sys.exit(1) 188 decode(input, output) 189 else: 190 if options.text: 191 if isinstance(input, str): 192 input = open(input, 'rb') 193 else: 194 print(sys.argv[0], ': cannot do -t from stdin') 195 sys.exit(1) 196 encode(input, output) 197 198 if __name__ == '__main__': 199 test() 200
