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 "libbb.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 int uuencode_main(int argc, char **argv)
  18 {
  19         const size_t src_buf_size = SRC_BUF_SIZE;
  20         const size_t dst_buf_size = DST_BUF_SIZE;
  21         size_t write_size = dst_buf_size;
  22         struct stat stat_buf;
  23         FILE *src_stream = stdin;
  24         const char *tbl;
  25         size_t size;
  26         mode_t mode;
  27         RESERVE_CONFIG_BUFFER(src_buf, SRC_BUF_SIZE + 1);
  28         RESERVE_CONFIG_BUFFER(dst_buf, DST_BUF_SIZE + 1);
  29
  30         tbl = bb_uuenc_tbl_std;
  31         if (getopt32(argc, argv, "m") & 1) {
  32                 tbl = bb_uuenc_tbl_base64;
  33         }
  34
  35         switch (argc - optind) {
  36                 case 2:
  37                         src_stream = xfopen(argv[optind], "r");
  38                         xstat(argv[optind], &stat_buf);
  39                         mode = stat_buf.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO);
  40                         if (src_stream == stdout) {
  41                                 puts("NULL");
  42                         }
  43                         break;
  44                 case 1:
  45                         mode = 0666 & ~umask(0666);
  46                         break;
  47                 default:
  48                         bb_show_usage();
  49         }
  50
  51         printf("begin%s %o %s", tbl == bb_uuenc_tbl_std ? "" : "-base64", mode, argv[argc - 1]);
  52
  53         while ((size = fread(src_buf, 1, src_buf_size, src_stream)) > 0) {
  54                 if (size != src_buf_size) {
  55                         /* write_size is always 60 until the last line */
  56                         write_size = (4 * ((size + 2) / 3));
  57                         /* pad with 0s so we can just encode extra bits */
  58                         memset(&src_buf[size], 0, src_buf_size - size);
  59                 }
  60                 /* Encode the buffer we just read in */
  61                 bb_uuencode((unsigned char*)src_buf, dst_buf, size, tbl);
  62
  63                 putchar('\n');
  64                 if (tbl == bb_uuenc_tbl_std) {
  65                         putchar(tbl[size]);
  66                 }
  67                 if (fwrite(dst_buf, 1, write_size, stdout) != write_size) {
  68                         bb_perror_msg_and_die(bb_msg_write_error);
  69                 }
  70         }
  71         printf(tbl == bb_uuenc_tbl_std ? "\n`\nend\n" : "\n====\n");
  72
  73         die_if_ferror(src_stream, "source");    /* TODO - Fix this! */
  74
  75         fflush_stdout_and_exit(EXIT_SUCCESS);
  76 }