"Robert P. J. Day" | 63fc1a9 | 2006-07-02 19:47:05 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 2 | /* |
Denis Vlasenko | 0beaff8 | 2007-09-21 13:16:32 +0000 | [diff] [blame] | 3 | * Copyright 2003, Glenn McGrath |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 4 | * |
Rob Landley | 0edbad1 | 2006-04-17 22:49:30 +0000 | [diff] [blame] | 5 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
Eric Andersen | 4e573f4 | 2000-11-14 23:29:24 +0000 | [diff] [blame] | 6 | * |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 7 | * Based on specification from |
| 8 | * http://www.opengroup.org/onlinepubs/007904975/utilities/uuencode.html |
Eric Andersen | 4e573f4 | 2000-11-14 23:29:24 +0000 | [diff] [blame] | 9 | * |
Rob Landley | 0edbad1 | 2006-04-17 22:49:30 +0000 | [diff] [blame] | 10 | * Bugs: the spec doesn't mention anything about "`\n`\n" prior to the |
| 11 | * "end" line |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 12 | */ |
| 13 | |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 14 | |
Denis Vlasenko | b6adbf1 | 2007-05-26 19:00:18 +0000 | [diff] [blame] | 15 | #include "libbb.h" |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 16 | |
Bernhard Reutner-Fischer | ca5b352 | 2007-01-20 21:29:32 +0000 | [diff] [blame] | 17 | static void read_stduu(FILE *src_stream, FILE *dst_stream) |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 18 | { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 19 | char *line; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 20 | |
Denis Vlasenko | 8ee649a | 2008-03-26 20:04:27 +0000 | [diff] [blame^] | 21 | while ((line = xmalloc_fgetline(src_stream)) != NULL) { |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 22 | int encoded_len, str_len; |
| 23 | char *line_ptr, *dst; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 24 | |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 25 | if (strcmp(line, "end") == 0) { |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 26 | return; /* the only non-error exit */ |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 27 | } |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 28 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 29 | line_ptr = line; |
| 30 | while (*line_ptr) { |
| 31 | *line_ptr = (*line_ptr - 0x20) & 0x3f; |
| 32 | line_ptr++; |
| 33 | } |
| 34 | str_len = line_ptr - line; |
| 35 | |
| 36 | encoded_len = line[0] * 4 / 3; |
| 37 | /* Check that line is not too short. (we tolerate |
| 38 | * overly _long_ line to accomodate possible extra '`'). |
| 39 | * Empty line case is also caught here. */ |
| 40 | if (str_len <= encoded_len) { |
| 41 | break; /* go to bb_error_msg_and_die("short file"); */ |
| 42 | } |
| 43 | if (encoded_len <= 0) { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 44 | /* Ignore the "`\n" line, why is it even in the encode file ? */ |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 45 | free(line); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 46 | continue; |
| 47 | } |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 48 | if (encoded_len > 60) { |
Denis Vlasenko | d3d004d | 2006-10-27 09:02:31 +0000 | [diff] [blame] | 49 | bb_error_msg_and_die("line too long"); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 50 | } |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 51 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 52 | dst = line; |
| 53 | line_ptr = line + 1; |
| 54 | do { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 55 | /* Merge four 6 bit chars to three 8 bit chars */ |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 56 | *dst++ = line_ptr[0] << 2 | line_ptr[1] >> 4; |
| 57 | encoded_len--; |
| 58 | if (encoded_len == 0) { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 59 | break; |
| 60 | } |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 61 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 62 | *dst++ = line_ptr[1] << 4 | line_ptr[2] >> 2; |
| 63 | encoded_len--; |
| 64 | if (encoded_len == 0) { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 65 | break; |
| 66 | } |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 67 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 68 | *dst++ = line_ptr[2] << 6 | line_ptr[3]; |
| 69 | line_ptr += 4; |
| 70 | encoded_len -= 2; |
| 71 | } while (encoded_len > 0); |
| 72 | fwrite(line, 1, dst - line, dst_stream); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 73 | free(line); |
| 74 | } |
Denis Vlasenko | d3d004d | 2006-10-27 09:02:31 +0000 | [diff] [blame] | 75 | bb_error_msg_and_die("short file"); |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 76 | } |
| 77 | |
Bernhard Reutner-Fischer | ca5b352 | 2007-01-20 21:29:32 +0000 | [diff] [blame] | 78 | static void read_base64(FILE *src_stream, FILE *dst_stream) |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 79 | { |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 80 | int term_count = 1; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 81 | |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 82 | while (1) { |
| 83 | char translated[4]; |
| 84 | int count = 0; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 85 | |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 86 | while (count < 4) { |
| 87 | char *table_ptr; |
Eric Andersen | 5e67887 | 2006-01-30 19:48:23 +0000 | [diff] [blame] | 88 | int ch; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 89 | |
Denis Vlasenko | aa198dd | 2007-06-12 07:24:11 +0000 | [diff] [blame] | 90 | /* Get next _valid_ character. |
| 91 | * global vector bb_uuenc_tbl_base64[] contains this string: |
| 92 | * "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=\n" |
| 93 | */ |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 94 | do { |
| 95 | ch = fgetc(src_stream); |
| 96 | if (ch == EOF) { |
Denis Vlasenko | d3d004d | 2006-10-27 09:02:31 +0000 | [diff] [blame] | 97 | bb_error_msg_and_die("short file"); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 98 | } |
Denis Vlasenko | aa198dd | 2007-06-12 07:24:11 +0000 | [diff] [blame] | 99 | table_ptr = strchr(bb_uuenc_tbl_base64, ch); |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 100 | } while (table_ptr == NULL); |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 101 | |
Denis Vlasenko | aa198dd | 2007-06-12 07:24:11 +0000 | [diff] [blame] | 102 | /* Convert encoded character to decimal */ |
| 103 | ch = table_ptr - bb_uuenc_tbl_base64; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 104 | |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 105 | if (*table_ptr == '=') { |
| 106 | if (term_count == 0) { |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 107 | translated[count] = '\0'; |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 108 | break; |
| 109 | } |
| 110 | term_count++; |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 111 | } else if (*table_ptr == '\n') { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 112 | /* Check for terminating line */ |
| 113 | if (term_count == 5) { |
Bernhard Reutner-Fischer | ca5b352 | 2007-01-20 21:29:32 +0000 | [diff] [blame] | 114 | return; |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 115 | } |
| 116 | term_count = 1; |
| 117 | continue; |
| 118 | } else { |
| 119 | translated[count] = ch; |
| 120 | count++; |
| 121 | term_count = 0; |
| 122 | } |
| 123 | } |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 124 | |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 125 | /* Merge 6 bit chars to 8 bit */ |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 126 | if (count > 1) { |
| 127 | fputc(translated[0] << 2 | translated[1] >> 4, dst_stream); |
| 128 | } |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 129 | if (count > 2) { |
Tim Riker | c1ef7bd | 2006-01-25 00:08:53 +0000 | [diff] [blame] | 130 | fputc(translated[1] << 4 | translated[2] >> 2, dst_stream); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 131 | } |
| 132 | if (count > 3) { |
Tim Riker | c1ef7bd | 2006-01-25 00:08:53 +0000 | [diff] [blame] | 133 | fputc(translated[2] << 6 | translated[3], dst_stream); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 134 | } |
| 135 | } |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 136 | } |
| 137 | |
Denis Vlasenko | 9b49a5e | 2007-10-11 10:05:36 +0000 | [diff] [blame] | 138 | int uudecode_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
Denis Vlasenko | 68404f1 | 2008-03-17 09:00:54 +0000 | [diff] [blame] | 139 | int uudecode_main(int argc ATTRIBUTE_UNUSED, char **argv) |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 140 | { |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 141 | FILE *src_stream; |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 142 | char *outname = NULL; |
| 143 | char *line; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 144 | |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 145 | opt_complementary = "?1"; /* 1 argument max */ |
Denis Vlasenko | fe7cd64 | 2007-08-18 15:32:12 +0000 | [diff] [blame] | 146 | getopt32(argv, "o:", &outname); |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 147 | argv += optind; |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 148 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 149 | if (!*argv) |
| 150 | *--argv = (char*)"-"; |
| 151 | src_stream = xfopen_stdin(*argv); |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 152 | |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 153 | /* Search for the start of the encoding */ |
Denis Vlasenko | 8ee649a | 2008-03-26 20:04:27 +0000 | [diff] [blame^] | 154 | while ((line = xmalloc_fgetline(src_stream)) != NULL) { |
Bernhard Reutner-Fischer | ca5b352 | 2007-01-20 21:29:32 +0000 | [diff] [blame] | 155 | void (*decode_fn_ptr)(FILE * src, FILE * dst); |
Rob Landley | 29d94b9 | 2006-09-23 19:56:21 +0000 | [diff] [blame] | 156 | char *line_ptr; |
| 157 | FILE *dst_stream; |
| 158 | int mode; |
Denis Vlasenko | f7996f3 | 2007-01-11 17:20:00 +0000 | [diff] [blame] | 159 | |
Rob Landley | 565bc33 | 2006-07-31 22:50:12 +0000 | [diff] [blame] | 160 | if (strncmp(line, "begin-base64 ", 13) == 0) { |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 161 | line_ptr = line + 13; |
| 162 | decode_fn_ptr = read_base64; |
| 163 | } else if (strncmp(line, "begin ", 6) == 0) { |
| 164 | line_ptr = line + 6; |
| 165 | decode_fn_ptr = read_stduu; |
Rob Landley | 29d94b9 | 2006-09-23 19:56:21 +0000 | [diff] [blame] | 166 | } else { |
| 167 | free(line); |
| 168 | continue; |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 169 | } |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 170 | |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 171 | /* begin line found. decode and exit */ |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 172 | mode = bb_strtou(line_ptr, NULL, 8); |
Rob Landley | 29d94b9 | 2006-09-23 19:56:21 +0000 | [diff] [blame] | 173 | if (outname == NULL) { |
| 174 | outname = strchr(line_ptr, ' '); |
| 175 | if ((outname == NULL) || (*outname == '\0')) { |
| 176 | break; |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 177 | } |
Rob Landley | 29d94b9 | 2006-09-23 19:56:21 +0000 | [diff] [blame] | 178 | outname++; |
| 179 | } |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 180 | dst_stream = stdout; |
| 181 | if (NOT_LONE_DASH(outname)) { |
Rob Landley | 29d94b9 | 2006-09-23 19:56:21 +0000 | [diff] [blame] | 182 | dst_stream = xfopen(outname, "w"); |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 183 | fchmod(fileno(dst_stream), mode & (S_IRWXU | S_IRWXG | S_IRWXO)); |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 184 | } |
| 185 | free(line); |
Bernhard Reutner-Fischer | ca5b352 | 2007-01-20 21:29:32 +0000 | [diff] [blame] | 186 | decode_fn_ptr(src_stream, dst_stream); |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 187 | /* fclose_if_not_stdin(src_stream); - redundant */ |
Bernhard Reutner-Fischer | ca5b352 | 2007-01-20 21:29:32 +0000 | [diff] [blame] | 188 | return EXIT_SUCCESS; |
Glenn L McGrath | 7f9de02 | 2003-11-06 03:17:23 +0000 | [diff] [blame] | 189 | } |
Denis Vlasenko | d3d004d | 2006-10-27 09:02:31 +0000 | [diff] [blame] | 190 | bb_error_msg_and_die("no 'begin' line"); |
Eric Andersen | 2b6ab3c | 2000-06-13 06:54:53 +0000 | [diff] [blame] | 191 | } |
Denis Vlasenko | 746204b | 2007-06-04 23:32:35 +0000 | [diff] [blame] | 192 | |
| 193 | /* Test script. |
| 194 | Put this into an empty dir with busybox binary, an run. |
| 195 | |
| 196 | #!/bin/sh |
| 197 | test -x busybox || { echo "No ./busybox?"; exit; } |
| 198 | ln -sf busybox uudecode |
| 199 | ln -sf busybox uuencode |
| 200 | >A_null |
| 201 | echo -n A >A |
| 202 | echo -n AB >AB |
| 203 | echo -n ABC >ABC |
| 204 | echo -n ABCD >ABCD |
| 205 | echo -n ABCDE >ABCDE |
| 206 | echo -n ABCDEF >ABCDEF |
| 207 | cat busybox >A_bbox |
| 208 | for f in A*; do |
| 209 | echo uuencode $f |
| 210 | ./uuencode $f <$f >u_$f |
| 211 | ./uuencode -m $f <$f >m_$f |
| 212 | done |
| 213 | mkdir unpk_u unpk_m 2>/dev/null |
| 214 | for f in u_*; do |
| 215 | ./uudecode <$f -o unpk_u/${f:2} |
| 216 | diff -a ${f:2} unpk_u/${f:2} >/dev/null 2>&1 |
| 217 | echo uudecode $f: $? |
| 218 | done |
| 219 | for f in m_*; do |
| 220 | ./uudecode <$f -o unpk_m/${f:2} |
| 221 | diff -a ${f:2} unpk_m/${f:2} >/dev/null 2>&1 |
| 222 | echo uudecode $f: $? |
| 223 | done |
| 224 | */ |