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  *  This program is free software; you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation; either version 2 of the License, or
  11  *  (at your option) any later version.
  12  *
  13  *  This program is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU Library General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with this program; if not, write to the Free Software
  20  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  21  */
  22 #include <getopt.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <sys/types.h>
  26 #include <sys/stat.h>
  27 #include <unistd.h>
  28 #include "busybox.h"
  29
  30 /* Conversion table.  for base 64 */
  31 static char tbl_base64[64] = {
  32         'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
  33         'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
  34         'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X',
  35         'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f',
  36         'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
  37         'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
  38         'w', 'x', 'y', 'z', '0', '1', '2', '3',
  39         '4', '5', '6', '7', '8', '9', '+', '/'
  40 };
  41
  42 static char tbl_std[64] = {
  43         '`', '!', '"', '#', '$', '%', '&', '\'',
  44         '(', ')', '*', '+', ',', '-', '.', '/',
  45         '0', '1', '2', '3', '4', '5', '6', '7',
  46         '8', '9', ':', ';', '<', '=', '>', '?',
  47         '@', 'A', 'B', 'C', 'D', 'E', 'F', 'G',
  48         'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O',
  49         'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W',
  50         'X', 'Y', 'Z', '[', '\\', ']', '^', '_'
  51 };
  52
  53 /*
  54  * Encode the string S of length LENGTH to base64 format and place it
  55  * to STORE.  STORE will be 0-terminated, and must point to a writable
  56  * buffer of at least 1+BASE64_LENGTH(length) bytes.
  57  * where BASE64_LENGTH(len) = (4 * ((LENGTH + 2) / 3))
  58  */
  59 static void uuencode (const char *s, const char *store, const int length, const char *tbl)
  60 {
  61         int i;
  62         unsigned char *p = (unsigned char *)store;
  63
  64         /* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
  65         for (i = 0; i < length; i += 3) {
  66                 *p++ = tbl[s[0] >> 2];
  67                 *p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
  68                 *p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
  69                 *p++ = tbl[s[2] & 0x3f];
  70                 s += 3;
  71         }
  72         /* Pad the result if necessary...  */
  73         if (i == length + 1) {
  74                 *(p - 1) = '=';
  75         }
  76         else if (i == length + 2) {
  77                 *(p - 1) = *(p - 2) = '=';
  78         }
  79         /* ...and zero-terminate it.  */
  80         *p = '\0';
  81 }
  82
  83 int uuencode_main(int argc, char **argv)
  84 {
  85         const int src_buf_size = 60;    // This *MUST* be a multiple of 3
  86         const int dst_buf_size = 4 * ((src_buf_size + 2) / 3);
  87         RESERVE_CONFIG_BUFFER(src_buf, src_buf_size + 1);
  88         RESERVE_CONFIG_BUFFER(dst_buf, dst_buf_size + 1);
  89         struct stat stat_buf;
  90         FILE *src_stream = stdin;
  91         char *tbl = tbl_std;
  92         size_t size;
  93         mode_t mode;
  94         int opt;
  95         int column = 0;
  96         int write_size = 0;
  97         int remaining;
  98         int buffer_offset = 0;
  99
 100         while ((opt = getopt(argc, argv, "m")) != -1) {
 101                 switch (opt) {
 102                 case 'm':
 103                         tbl = tbl_base64;
 104                         break;
 105                 default:
 106                         show_usage();
 107                 }
 108         }
 109
 110         switch (argc - optind) {
 111                 case 2:
 112                         src_stream = xfopen(argv[optind], "r");
 113                         stat(argv[optind], &stat_buf);
 114                         mode = stat_buf.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO);
 115                         if (src_stream == stdout) {
 116                                 printf("NULL\n");
 117                         }
 118                         break;
 119                 case 1:
 120                         mode = 0666 & ~umask(0666);
 121                         break;
 122                 default:
 123                         show_usage();
 124         }
 125
 126         printf("begin%s %o %s", tbl == tbl_std ? "" : "-base64", mode, argv[argc - 1]);
 127
 128         while ((size = fread(src_buf, 1, src_buf_size, src_stream)) > 0) {
 129                 /* Encode the buffer we just read in */
 130                 uuencode(src_buf, dst_buf, size, tbl);
 131
 132                 /* Write the buffer to stdout, wrapping at 60 chars.
 133                  * This looks overly complex, but it gets tricky as
 134                  * the line has to continue to wrap correctly if we
 135                  * have to refill the buffer
 136                  *
 137                  * Improvments most welcome
 138                  */
 139
 140                 /* Initialise values for the new buffer */
 141                 remaining = 4 * ((size + 2) / 3);
 142                 buffer_offset = 0;
 143
 144                 /* Write the buffer to stdout, wrapping at 60 chars
 145                  * starting from the column the last buffer ran out
 146                  */
 147                 do {
 148                         if (remaining > (60 - column)) {
 149                                 write_size = 60 - column;
 150                         }
 151                         else if (remaining < 60) {
 152                                 write_size = remaining;
 153                         } else {
 154                                 write_size = 60;
 155                         }
 156
 157                         /* Setup a new row if required */
 158                         if (column == 0) {
 159                                 putchar('\n');
 160                                 if (tbl == tbl_std) {
 161                                         putchar('M');
 162                                 }
 163                         }
 164                         /* Write to the 60th column */
 165                         if (fwrite(&dst_buf[buffer_offset], 1, write_size, stdout) != write_size) {
 166                                 perror("Couldnt finish writing");
 167                         }
 168                         /* Update variables based on last write */
 169                         buffer_offset += write_size;
 170                         remaining -= write_size;
 171                         column += write_size;
 172                         if (column % 60 == 0) {
 173                                 column = 0;
 174                         }
 175                 } while (remaining > 0);
 176         }
 177         printf(tbl == tbl_std ? "\n`\nend\n" : "\n====\n");
 178
 179         return(EXIT_SUCCESS);
 180 }