#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);
}
};
/*
- * Encode the string S of length LENGTH to base64 format and place it
+ * Encode bytes at S of length LENGTH to uuencode or base64 format and place it
* to STORE. STORE will be 0-terminated, and must point to a writable
* buffer of at least 1+BASE64_LENGTH(length) bytes.
* where BASE64_LENGTH(len) = (4 * ((LENGTH + 2) / 3))
*/
-void bb_uuencode(const unsigned char *s, char *store, const int length, const char *tbl)
+void bb_uuencode(char *store, const void *src, int length, const char *tbl)
{
int i;
+ const unsigned char *s = src;
char *p = store;
/* Transform the 3x8 bits to 4x6 bits, as required by base64. */