introduce LONE_CHAR (optimized strcmp with one-char string)
[oweals/busybox.git] / coreutils / uuencode.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  *  Copyright (C) 2000 by Glenn McGrath
4  *
5  *  based on the function base64_encode from http.c in wget v1.6
6  *  Copyright (C) 1995, 1996, 1997, 1998, 2000 Free Software Foundation, Inc.
7  *
8  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
9  */
10
11 #include "busybox.h"
12
13
14 #define SRC_BUF_SIZE    45  // This *MUST* be a multiple of 3
15 #define DST_BUF_SIZE    4 * ((SRC_BUF_SIZE + 2) / 3)
16 int uuencode_main(int argc, char **argv)
17 {
18         const size_t src_buf_size = SRC_BUF_SIZE;
19         const size_t dst_buf_size = DST_BUF_SIZE;
20         size_t write_size = dst_buf_size;
21         struct stat stat_buf;
22         FILE *src_stream = stdin;
23         const char *tbl;
24         size_t size;
25         mode_t mode;
26         RESERVE_CONFIG_BUFFER(src_buf, SRC_BUF_SIZE + 1);
27         RESERVE_CONFIG_BUFFER(dst_buf, DST_BUF_SIZE + 1);
28
29         tbl = bb_uuenc_tbl_std;
30         if (getopt32(argc, argv, "m") & 1) {
31                 tbl = bb_uuenc_tbl_base64;
32         }
33
34         switch (argc - optind) {
35                 case 2:
36                         src_stream = xfopen(argv[optind], "r");
37                         xstat(argv[optind], &stat_buf);
38                         mode = stat_buf.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO);
39                         if (src_stream == stdout) {
40                                 puts("NULL");
41                         }
42                         break;
43                 case 1:
44                         mode = 0666 & ~umask(0666);
45                         break;
46                 default:
47                         bb_show_usage();
48         }
49
50         printf("begin%s %o %s", tbl == bb_uuenc_tbl_std ? "" : "-base64", mode, argv[argc - 1]);
51
52         while ((size = fread(src_buf, 1, src_buf_size, src_stream)) > 0) {
53                 if (size != src_buf_size) {
54                         /* write_size is always 60 until the last line */
55                         write_size = (4 * ((size + 2) / 3));
56                         /* pad with 0s so we can just encode extra bits */
57                         memset(&src_buf[size], 0, src_buf_size - size);
58                 }
59                 /* Encode the buffer we just read in */
60                 bb_uuencode((unsigned char*)src_buf, dst_buf, size, tbl);
61
62                 putchar('\n');
63                 if (tbl == bb_uuenc_tbl_std) {
64                         putchar(tbl[size]);
65                 }
66                 if (fwrite(dst_buf, 1, write_size, stdout) != write_size) {
67                         bb_perror_msg_and_die(bb_msg_write_error);
68                 }
69         }
70         printf(tbl == bb_uuenc_tbl_std ? "\n`\nend\n" : "\n====\n");
71
72         die_if_ferror(src_stream, "source");    /* TODO - Fix this! */
73
74         fflush_stdout_and_exit(EXIT_SUCCESS);
75 }