1 /* $OpenBSD: uuencode.c,v 1.10 2009/10/27 23:59:46 deraadt Exp $ */ 2 /* $FreeBSD: uuencode.c,v 1.18 2004/01/22 07:23:35 grehan Exp $ */ 3 4 /*- 5 * Copyright (c) 1983, 1993 6 * The Regents of the University of California. All rights reserved. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 1. Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. Neither the name of the University nor the names of its contributors 17 * may be used to endorse or promote products derived from this software 18 * without specific prior written permission. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 23 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 */ 32 33 /* 34 * Encode a file so it can be mailed to a remote system. 35 */ 36 37 #include <sys/param.h> 38 #include <sys/socket.h> 39 #include <sys/stat.h> 40 41 #include <netinet/in.h> 42 43 #include <err.h> 44 #include <locale.h> 45 #include <resolv.h> 46 #include <stdio.h> 47 #include <stdlib.h> 48 #include <string.h> 49 #include <unistd.h> 50 51 void encode(void); 52 void base64_encode(void); 53 static void usage(void); 54 55 FILE *output; 56 int mode; 57 char **av; 58 59 enum program_mode { 60 MODE_ENCODE, 61 MODE_B64ENCODE 62 } pmode; 63 64 int 65 main(int argc, char *argv[]) 66 { 67 struct stat sb; 68 int base64, ch; 69 char *outfile; 70 extern char *__progname; 71 static const char *optstr[2] = { 72 "mo:", 73 "o:" 74 }; 75 76 base64 = 0; 77 outfile = NULL; 78 79 pmode = MODE_ENCODE; 80 if (strcmp(__progname, "b64encode") == 0) { 81 base64 = 1; 82 pmode = MODE_B64ENCODE; 83 } 84 85 setlocale(LC_ALL, ""); 86 while ((ch = getopt(argc, argv, optstr[pmode])) != -1) { 87 switch (ch) { 88 case 'm': 89 base64 = 1; 90 break; 91 case 'o': 92 outfile = optarg; 93 break; 94 case '?': 95 default: 96 usage(); 97 } 98 } 99 argv += optind; 100 argc -= optind; 101 102 switch(argc) { 103 case 2: /* optional first argument is input file */ 104 if (!freopen(*argv, "r", stdin) || fstat(fileno(stdin), &sb)) 105 err(1, "%s", *argv); 106 #define RWX (S_IRWXU|S_IRWXG|S_IRWXO) 107 mode = sb.st_mode & RWX; 108 ++argv; 109 break; 110 case 1: 111 #define RW (S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH) 112 mode = RW & ~umask(RW); 113 break; 114 case 0: 115 default: 116 usage(); 117 } 118 119 av = argv; 120 121 if (outfile != NULL) { 122 output = fopen(outfile, "w+"); 123 if (output == NULL) 124 err(1, "unable to open %s for output", outfile); 125 } else 126 output = stdout; 127 if (base64) 128 base64_encode(); 129 else 130 encode(); 131 if (ferror(output)) 132 errx(1, "write error"); 133 exit(0); 134 } 135 136 /* ENC is the basic 1 character encoding function to make a char printing */ 137 #define ENC(c) ((c) ? ((c) & 077) + ' ': '`') 138 139 /* 140 * Copy from in to out, encoding in base64 as you go along. 141 */ 142 void 143 base64_encode(void) 144 { 145 /* 146 * Output must fit into 80 columns, chunks come in 4, leave 1. 147 */ 148 #define GROUPS ((80 / 4) - 1) 149 unsigned char buf[3]; 150 char buf2[sizeof(buf) * 2 + 1]; 151 size_t n; 152 int rv, sequence; 153 154 sequence = 0; 155 156 fprintf(output, "begin-base64 %o %s\n", mode, *av); 157 while ((n = fread(buf, 1, sizeof(buf), stdin))) { 158 ++sequence; 159 rv = b64_ntop(buf, n, buf2, (sizeof(buf2) / sizeof(buf2[0]))); 160 if (rv == -1) 161 errx(1, "b64_ntop: error encoding base64"); 162 fprintf(output, "%s%s", buf2, (sequence % GROUPS) ? "" : "\n"); 163 } 164 if (sequence % GROUPS) 165 fprintf(output, "\n"); 166 fprintf(output, "====\n"); 167 } 168 169 /* 170 * Copy from in to out, encoding as you go along. 171 */ 172 void 173 encode(void) 174 { 175 int ch, n; 176 char *p; 177 char buf[80]; 178 179 (void)fprintf(output, "begin %o %s\n", mode, *av); 180 while ((n = fread(buf, 1, 45, stdin))) { 181 ch = ENC(n); 182 if (fputc(ch, output) == EOF) 183 break; 184 for (p = buf; n > 0; n -= 3, p += 3) { 185 /* Pad with nulls if not a multiple of 3. */ 186 if (n < 3) { 187 p[2] = '\0'; 188 if (n < 2) 189 p[1] = '\0'; 190 } 191 ch = *p >> 2; 192 ch = ENC(ch); 193 if (fputc(ch, output) == EOF) 194 break; 195 ch = ((*p << 4) & 060) | ((p[1] >> 4) & 017); 196 ch = ENC(ch); 197 if (fputc(ch, output) == EOF) 198 break; 199 ch = ((p[1] << 2) & 074) | ((p[2] >> 6) & 03); 200 ch = ENC(ch); 201 if (fputc(ch, output) == EOF) 202 break; 203 ch = p[2] & 077; 204 ch = ENC(ch); 205 if (fputc(ch, output) == EOF) 206 break; 207 } 208 if (fputc('\n', output) == EOF) 209 break; 210 } 211 if (ferror(stdin)) 212 errx(1, "read error"); 213 (void)fprintf(output, "%c\nend\n", ENC('\0')); 214 } 215 216 static void 217 usage(void) 218 { 219 switch (pmode) { 220 case MODE_ENCODE: 221 (void)fprintf(stderr, 222 "usage: uuencode [-m] [-o output_file] [file] name\n"); 223 break; 224 case MODE_B64ENCODE: 225 (void)fprintf(stderr, 226 "usage: b64encode [-o output_file] [file] name\n"); 227 break; 228 } 229 exit(1); 230 } 231