1 /* 2 * ***************************************************************************** 3 * 4 * SPDX-License-Identifier: BSD-2-Clause 5 * 6 * Copyright (c) 2018-2020 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 <status.h> 45 #include <vector.h> 46 #include <read.h> 47 #include <vm.h> 48 #include <args.h> 49 #include <opt.h> 50 51 static const BcOptLong bc_args_lopt[] = { 52 53 { "expression", BC_OPT_REQUIRED, 'e' }, 54 { "file", BC_OPT_REQUIRED, 'f' }, 55 { "help", BC_OPT_NONE, 'h' }, 56 { "interactive", BC_OPT_NONE, 'i' }, 57 { "no-prompt", BC_OPT_NONE, 'P' }, 58 #if BC_ENABLED 59 { "global-stacks", BC_OPT_BC_ONLY, 'g' }, 60 { "mathlib", BC_OPT_BC_ONLY, 'l' }, 61 { "quiet", BC_OPT_BC_ONLY, 'q' }, 62 { "standard", BC_OPT_BC_ONLY, 's' }, 63 { "warn", BC_OPT_BC_ONLY, 'w' }, 64 #endif // BC_ENABLED 65 { "version", BC_OPT_NONE, 'v' }, 66 { "version", BC_OPT_NONE, 'V' }, 67 #if DC_ENABLED 68 { "extended-register", BC_OPT_DC_ONLY, 'x' }, 69 #endif // DC_ENABLED 70 { NULL, 0, 0 }, 71 72 }; 73 74 static void bc_args_exprs(const char *str) { 75 BC_SIG_ASSERT_LOCKED; 76 if (vm.exprs.v == NULL) bc_vec_init(&vm.exprs, sizeof(uchar), NULL); 77 bc_vec_concat(&vm.exprs, str); 78 bc_vec_concat(&vm.exprs, "\n"); 79 } 80 81 static void bc_args_file(const char *file) { 82 83 char *buf; 84 85 BC_SIG_ASSERT_LOCKED; 86 87 vm.file = file; 88 89 bc_read_file(file, &buf); 90 bc_args_exprs(buf); 91 free(buf); 92 } 93 94 void bc_args(int argc, char *argv[]) { 95 96 int c; 97 size_t i; 98 bool do_exit = false, version = false; 99 BcOpt opts; 100 101 BC_SIG_ASSERT_LOCKED; 102 103 bc_opt_init(&opts, argv); 104 105 while ((c = bc_opt_parse(&opts, bc_args_lopt)) != -1) { 106 107 switch (c) { 108 109 case 'e': 110 { 111 if (vm.no_exit_exprs) 112 bc_vm_verr(BC_ERROR_FATAL_OPTION, "-e (--expression)"); 113 bc_args_exprs(opts.optarg); 114 break; 115 } 116 117 case 'f': 118 { 119 if (!strcmp(opts.optarg, "-")) vm.no_exit_exprs = true; 120 else { 121 if (vm.no_exit_exprs) 122 bc_vm_verr(BC_ERROR_FATAL_OPTION, "-f (--file)"); 123 bc_args_file(opts.optarg); 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