1 /* 2 * ***************************************************************************** 3 * 4 * SPDX-License-Identifier: BSD-2-Clause 5 * 6 * Copyright (c) 2018-2021 Gavin D. Howard and contributors. 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions are met: 10 * 11 * * Redistributions of source code must retain the above copyright notice, this 12 * list of conditions and the following disclaimer. 13 * 14 * * Redistributions in binary form must reproduce the above copyright notice, 15 * this list of conditions and the following disclaimer in the documentation 16 * and/or other materials provided with the distribution. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 19 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 21 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE 22 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 28 * POSSIBILITY OF SUCH DAMAGE. 29 * 30 * ***************************************************************************** 31 * 32 * Code for processing command-line arguments. 33 * 34 */ 35 36 #include <assert.h> 37 #include <ctype.h> 38 #include <stdbool.h> 39 #include <stdlib.h> 40 #include <string.h> 41 42 #include <unistd.h> 43 44 #include <vector.h> 45 #include <read.h> 46 #include <args.h> 47 #include <opt.h> 48 49 static const BcOptLong bc_args_lopt[] = { 50 51 { "expression", BC_OPT_REQUIRED, 'e' }, 52 { "file", BC_OPT_REQUIRED, 'f' }, 53 { "help", BC_OPT_NONE, 'h' }, 54 { "interactive", BC_OPT_NONE, 'i' }, 55 { "no-prompt", BC_OPT_NONE, 'P' }, 56 #if BC_ENABLED 57 { "global-stacks", BC_OPT_BC_ONLY, 'g' }, 58 { "mathlib", BC_OPT_BC_ONLY, 'l' }, 59 { "quiet", BC_OPT_BC_ONLY, 'q' }, 60 { "standard", BC_OPT_BC_ONLY, 's' }, 61 { "warn", BC_OPT_BC_ONLY, 'w' }, 62 #endif // BC_ENABLED 63 { "version", BC_OPT_NONE, 'v' }, 64 { "version", BC_OPT_NONE, 'V' }, 65 #if DC_ENABLED 66 { "extended-register", BC_OPT_DC_ONLY, 'x' }, 67 #endif // DC_ENABLED 68 { NULL, 0, 0 }, 69 70 }; 71 72 static void bc_args_exprs(const char *str) { 73 BC_SIG_ASSERT_LOCKED; 74 if (vm.exprs.v == NULL) bc_vec_init(&vm.exprs, sizeof(uchar), NULL); 75 bc_vec_concat(&vm.exprs, str); 76 bc_vec_concat(&vm.exprs, "\n"); 77 } 78 79 static void bc_args_file(const char *file) { 80 81 char *buf; 82 83 BC_SIG_ASSERT_LOCKED; 84 85 vm.file = file; 86 87 bc_read_file(file, &buf); 88 bc_args_exprs(buf); 89 free(buf); 90 } 91 92 void bc_args(int argc, char *argv[], bool exit_exprs) { 93 94 int c; 95 size_t i; 96 bool do_exit = false, version = false; 97 BcOpt opts; 98 99 BC_SIG_ASSERT_LOCKED; 100 101 bc_opt_init(&opts, argv); 102 103 while ((c = bc_opt_parse(&opts, bc_args_lopt)) != -1) { 104 105 switch (c) { 106 107 case 'e': 108 { 109 if (vm.no_exit_exprs) 110 bc_vm_verr(BC_ERR_FATAL_OPTION, "-e (--expression)"); 111 bc_args_exprs(opts.optarg); 112 vm.exit_exprs = (exit_exprs || vm.exit_exprs); 113 break; 114 } 115 116 case 'f': 117 { 118 if (!strcmp(opts.optarg, "-")) vm.no_exit_exprs = true; 119 else { 120 if (vm.no_exit_exprs) 121 bc_vm_verr(BC_ERR_FATAL_OPTION, "-f (--file)"); 122 bc_args_file(opts.optarg); 123 vm.exit_exprs = (exit_exprs || vm.exit_exprs); 124 } 125 break; 126 } 127 128 case 'h': 129 { 130 bc_vm_info(vm.help); 131 do_exit = true; 132 break; 133 } 134 135 case 'i': 136 { 137 vm.flags |= BC_FLAG_I; 138 break; 139 } 140 141 case 'P': 142 { 143 vm.flags |= BC_FLAG_P; 144 break; 145 } 146 147 #if BC_ENABLED 148 case 'g': 149 { 150 assert(BC_IS_BC); 151 vm.flags |= BC_FLAG_G; 152 break; 153 } 154 155 case 'l': 156 { 157 assert(BC_IS_BC); 158 vm.flags |= BC_FLAG_L; 159 break; 160 } 161 162 case 'q': 163 { 164 assert(BC_IS_BC); 165 // Do nothing. 166 break; 167 } 168 169 case 's': 170 { 171 assert(BC_IS_BC); 172 vm.flags |= BC_FLAG_S; 173 break; 174 } 175 176 case 'w': 177 { 178 assert(BC_IS_BC); 179 vm.flags |= BC_FLAG_W; 180 break; 181 } 182 #endif // BC_ENABLED 183 184 case 'V': 185 case 'v': 186 { 187 do_exit = version = true; 188 break; 189 } 190 191 #if DC_ENABLED 192 case 'x': 193 { 194 assert(BC_IS_DC); 195 vm.flags |= DC_FLAG_X; 196 break; 197 } 198 #endif // DC_ENABLED 199 200 #ifndef NDEBUG 201 // We shouldn't get here because bc_opt_error()/bc_vm_error() should 202 // longjmp() out. 203 case '?': 204 case ':': 205 default: 206 { 207 abort(); 208 } 209 #endif // NDEBUG 210 } 211 } 212 213 if (version) bc_vm_info(NULL); 214 if (do_exit) exit((int) vm.status); 215 216 if (opts.optind < (size_t) argc && vm.files.v == NULL) 217 bc_vec_init(&vm.files, sizeof(char*), NULL); 218 219 for (i = opts.optind; i < (size_t) argc; ++i) 220 bc_vec_push(&vm.files, argv + i); 221 } 222