1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause 3 * 4 * Copyright (c) 2022 The FreeBSD Foundation 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #include <sys/param.h> 29 #include <getopt.h> 30 #include <libgen.h> 31 #include <stdbool.h> 32 #include <stdio.h> 33 #include <stdlib.h> 34 #include <string.h> 35 #include <sysexits.h> 36 37 extern int main_decode(int, char *[]); 38 extern int main_encode(int, char *[]); 39 extern int main_base64_decode(const char *); 40 extern int main_base64_encode(const char *, const char *); 41 42 static int search(const char *const); 43 static void usage_base64(bool); 44 static void version_base64(void); 45 static void base64_encode_or_decode(int, char *[]); 46 47 enum coders { 48 uuencode, uudecode, b64encode, b64decode, base64 49 }; 50 51 int 52 main(int argc, char *argv[]) 53 { 54 const char *const progname = getprogname(); 55 int coder = search(progname); 56 57 if (coder == -1 && argc > 1) { 58 argc--; 59 argv++; 60 coder = search(argv[0]); 61 } 62 switch (coder) { 63 case uuencode: 64 case b64encode: 65 main_encode(argc, argv); 66 break; 67 case uudecode: 68 case b64decode: 69 main_decode(argc, argv); 70 break; 71 case base64: 72 base64_encode_or_decode(argc, argv); 73 break; 74 default: 75 (void)fprintf(stderr, 76 "usage: %1$s <uuencode | uudecode> ...\n" 77 " %1$s <b64encode | b64decode> ...\n" 78 " %1$s <base64> ...\n", 79 progname); 80 exit(EX_USAGE); 81 } 82 } 83 84 static int 85 search(const char *const progname) 86 { 87 #define DESIGNATE(item) [item] = #item 88 const char *const known[] = { 89 DESIGNATE(uuencode), 90 DESIGNATE(uudecode), 91 DESIGNATE(b64encode), 92 DESIGNATE(b64decode), 93 DESIGNATE(base64) 94 }; 95 96 for (size_t i = 0; i < nitems(known); i++) 97 if (strcmp(progname, known[i]) == 0) 98 return ((int)i); 99 return (-1); 100 } 101 102 static void 103 usage_base64(bool failure) 104 { 105 (void)fputs("usage: base64 [-w col | --wrap=col] " 106 "[-d | --decode] [FILE]\n" 107 " base64 --help\n" 108 " base64 --version\n", stderr); 109 exit(failure ? EXIT_FAILURE : EXIT_SUCCESS); 110 } 111 112 static void 113 version_base64(void) 114 { 115 (void)fputs("FreeBSD base64\n", stderr); 116 exit(EXIT_SUCCESS); 117 } 118 119 static void 120 base64_encode_or_decode(int argc, char *argv[]) 121 { 122 int ch; 123 bool decode = false; 124 const char *w = NULL; 125 enum { HELP, VERSION }; 126 static const struct option opts[] = 127 { 128 {"decode", no_argument, NULL, 'd'}, 129 {"ignore-garbage",no_argument, NULL, 'i'}, 130 {"wrap", required_argument, NULL, 'w'}, 131 {"help", no_argument, NULL, HELP}, 132 {"version", no_argument, NULL, VERSION}, 133 {NULL, no_argument, NULL, 0} 134 }; 135 136 while ((ch = getopt_long(argc, argv, "diw:", opts, NULL)) != -1) 137 switch (ch) { 138 case 'd': 139 decode = true; 140 break; 141 case 'w': 142 w = optarg; 143 break; 144 case VERSION: 145 version_base64(); 146 case HELP: 147 default: 148 usage_base64(ch == '?'); 149 } 150 151 if (decode) 152 main_base64_decode(argv[optind]); 153 else 154 main_base64_encode(argv[optind], w); 155 } 156