X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=coreutils%2Fuuencode.c;h=e19f996764667f9bf5470e67377577bf187c4ead;hb=8d680b51148b9cbe7dd3afd37022f3390dd999ef;hp=1a8882fc7e5942f171ac2cf3a6f68908d3407dba;hpb=21afc7dc291f1cb11feec7a9766bf3542545f581;p=oweals%2Fbusybox.git

diff --git a/coreutils/uuencode.c b/coreutils/uuencode.c
index 1a8882fc7..e19f99676 100644
--- a/coreutils/uuencode.c
+++ b/coreutils/uuencode.c
@@ -8,68 +8,54 @@
  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 
-#include "busybox.h"
+#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) MAIN_EXTERNALLY_VISIBLE;
 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;
-	if (bb_getopt_ulflags(argc, argv, "m") & 1) {
+	mode = 0666 & ~umask(0666);
+	opt_complementary = "-1:?2"; /* must have 1 or 2 args */
+	if (getopt32(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);
-			if (src_stream == stdout) {
-				puts("NULL");
-			}
-			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++;
 	}
 
-	bb_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));
 	}
-	bb_printf(tbl == bb_uuenc_tbl_std ? "\n`\nend\n" : "\n====\n");
-
-	xferror(src_stream, "source");	/* TODO - Fix this! */
+	printf(tbl == bb_uuenc_tbl_std ? "\n`\nend\n" : "\n====\n");
 
-	bb_fflush_stdout_and_exit(EXIT_SUCCESS);
+	fflush_stdout_and_exit(EXIT_SUCCESS);
 }