hush: if we did match "LINENO" or "OPTIND", stop further comparisons
[oweals/busybox.git] / coreutils / uuencode.c
index b54e317a2744efd11d5b163698fe89080bf878ff..aa53b14a64fb10421f3ae6b001a4c79663bdd2f0 100644 (file)
@@ -1,73 +1,82 @@
 /* vi: set sw=4 ts=4: */
 /*
- *  Copyright (C) 2000 by Glenn McGrath
+ * Copyright (C) 2000 by Glenn McGrath
  *
- *  based on the function base64_encode from http.c in wget v1.6
- *  Copyright (C) 1995, 1996, 1997, 1998, 2000 Free Software Foundation, Inc.
+ * based on the function base64_encode from http.c in wget v1.6
+ * Copyright (C) 1995, 1996, 1997, 1998, 2000 Free Software Foundation, Inc.
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
+//config:config UUENCODE
+//config:      bool "uuencode (4.6 kb)"
+//config:      default y
+//config:      help
+//config:      uuencode is used to uuencode a file.
+
+//applet:IF_UUENCODE(APPLET(uuencode, BB_DIR_USR_BIN, BB_SUID_DROP))
+
+//kbuild:lib-$(CONFIG_UUENCODE) += uuencode.o
+
+//usage:#define uuencode_trivial_usage
+//usage:       "[-m] [FILE] STORED_FILENAME"
+//usage:#define uuencode_full_usage "\n\n"
+//usage:       "Uuencode FILE (or stdin) to stdout\n"
+//usage:     "\n       -m      Use base64 encoding per RFC1521"
+//usage:
+//usage:#define uuencode_example_usage
+//usage:       "$ uuencode busybox busybox\n"
+//usage:       "begin 755 busybox\n"
+//usage:       "<encoded file snipped>\n"
+//usage:       "$ uudecode busybox busybox > busybox.uu\n"
+//usage:       "$\n"
 
 #include "libbb.h"
 
+enum {
+       SRC_BUF_SIZE = 15*3,  /* 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)
+int uuencode_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int uuencode_main(int argc UNUSED_PARAM, 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;
-       if (getopt32(argc, argv, "m")) {
+       mode = 0666 & ~umask(0666);
+       if (getopt32(argv, "^" "m" "\0" "-1:?2"/*must have 1 or 2 args*/)) {
                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 (argv[1]) {
+               src_fd = xopen(argv[0], 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);
-
-               putchar('\n');
+               bb_uuencode(dst_buf, src_buf, size, tbl);
+               bb_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);
+                       bb_putchar(tbl[size]);
                }
+               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);
 }