| @@ -10,64 +10,52 @@ |
| |
| #include "libbb.h" |
| |
| + | enum { |
| + | SRC_BUF_SIZE = 45, /* This *MUST* be a multiple of 3 */ |
| + | DST_BUF_SIZE = 4 * ((SRC_BUF_SIZE + 2) / 3), |
| + | }; |
| |
| - | #define SRC_BUF_SIZE 45 // This *MUST* be a multiple of 3 |
| - | #define DST_BUF_SIZE 4 * ((SRC_BUF_SIZE + 2) / 3) |
| int uuencode_main(int argc, char **argv); |
| int uuencode_main(int argc, char **argv) |
| { |
| - | const size_t src_buf_size = SRC_BUF_SIZE; |
| - | const size_t dst_buf_size = DST_BUF_SIZE; |
| - | size_t write_size = dst_buf_size; |
| struct stat stat_buf; |
| - | FILE *src_stream = stdin; |
| + | int src_fd = STDIN_FILENO; |
| const char *tbl; |
| - | size_t size; |
| mode_t mode; |
| - | RESERVE_CONFIG_BUFFER(src_buf, SRC_BUF_SIZE + 1); |
| - | RESERVE_CONFIG_BUFFER(dst_buf, DST_BUF_SIZE + 1); |
| + | char src_buf[SRC_BUF_SIZE]; |
| + | char dst_buf[DST_BUF_SIZE + 1]; |
| |
| tbl = bb_uuenc_tbl_std; |
| + | mode = 0666 & ~umask(0666); |
| + | opt_complementary = "-1:?2"; /* must have 1 or 2 args */ |
| if (getopt32(argc, argv, "m")) { |
| tbl = bb_uuenc_tbl_base64; |
| } |
| - | |
| - | switch (argc - optind) { |
| - | case 2: |
| - | src_stream = xfopen(argv[optind], "r"); |
| - | xstat(argv[optind], &stat_buf); |
| - | mode = stat_buf.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO); |
| - | break; |
| - | case 1: |
| - | mode = 0666 & ~umask(0666); |
| - | break; |
| - | default: |
| - | bb_show_usage(); |
| + | argv += optind; |
| + | if (argc == optind + 2) { |
| + | src_fd = xopen(*argv, O_RDONLY); |
| + | fstat(src_fd, &stat_buf); |
| + | mode = stat_buf.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO); |
| + | argv++; |
| } |
| |
| - | printf("begin%s %o %s", tbl == bb_uuenc_tbl_std ? "" : "-base64", mode, argv[argc - 1]); |
| - | |
| - | while ((size = fread(src_buf, 1, src_buf_size, src_stream)) > 0) { |
| - | if (size != src_buf_size) { |
| - | /* write_size is always 60 until the last line */ |
| - | write_size = (4 * ((size + 2) / 3)); |
| - | /* pad with 0s so we can just encode extra bits */ |
| - | memset(&src_buf[size], 0, src_buf_size - size); |
| - | } |
| + | printf("begin%s %o %s", tbl == bb_uuenc_tbl_std ? "" : "-base64", mode, *argv); |
| + | while (1) { |
| + | size_t size = full_read(src_fd, src_buf, SRC_BUF_SIZE); |
| + | if (!size) |
| + | break; |
| + | if ((ssize_t)size < 0) |
| + | bb_perror_msg_and_die(bb_msg_read_error); |
| /* Encode the buffer we just read in */ |
| - | bb_uuencode((unsigned char*)src_buf, dst_buf, size, tbl); |
| - | |
| + | bb_uuencode(dst_buf, src_buf, size, tbl); |
| putchar('\n'); |
| if (tbl == bb_uuenc_tbl_std) { |
| putchar(tbl[size]); |
| } |
| - | if (fwrite(dst_buf, 1, write_size, stdout) != write_size) { |
| - | bb_perror_msg_and_die(bb_msg_write_error); |
| - | } |
| + | fflush(stdout); |
| + | xwrite(STDOUT_FILENO, dst_buf, 4 * ((size + 2) / 3)); |
| } |
| printf(tbl == bb_uuenc_tbl_std ? "\n`\nend\n" : "\n====\n"); |
| |
| - | die_if_ferror(src_stream, "source"); /* TODO - Fix this! */ |
| - | |
| fflush_stdout_and_exit(EXIT_SUCCESS); |
| } |