2 * *****************************************************************************
4 * SPDX-License-Identifier: BSD-2-Clause
6 * Copyright (c) 2018-2021 Gavin D. Howard and contributors.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are met:
11 * * Redistributions of source code must retain the above copyright notice, this
12 * list of conditions and the following disclaimer.
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.
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.
30 * *****************************************************************************
32 * Code for processing command-line arguments.
52 * Adds @a str to the list of expressions to execute later.
53 * @param str The string to add to the list of expressions.
55 static void bc_args_exprs(const char *str) {
57 if (vm.exprs.v == NULL) bc_vec_init(&vm.exprs, sizeof(uchar), BC_DTOR_NONE);
58 bc_vec_concat(&vm.exprs, str);
59 bc_vec_concat(&vm.exprs, "\n");
63 * Adds the contents of @a file to the list of expressions to execute later.
64 * @param file The name of the file whose contents should be added to the list
65 * of expressions to execute.
67 static void bc_args_file(const char *file) {
75 buf = bc_read_file(file);
86 * Redefines a keyword, if it exists and is not a POSIX keyword. Otherwise, it
87 * throws a fatal error.
88 * @param keyword The keyword to redefine.
90 static void bc_args_redefine(const char *keyword) {
96 for (i = 0; i < bc_lex_kws_len; ++i) {
98 const BcLexKeyword *kw = bc_lex_kws + i;
100 if (!strcmp(keyword, kw->name)) {
102 if (BC_LEX_KW_POSIX(kw)) break;
104 vm.redefined_kws[i] = true;
110 bc_error(BC_ERR_FATAL_ARG, 0, keyword);
115 void bc_args(int argc, char *argv[], bool exit_exprs) {
119 bool do_exit = false, version = false;
122 BC_SIG_ASSERT_LOCKED;
124 bc_opt_init(&opts, argv);
126 // This loop should look familiar to anyone who has used getopt() or
127 // getopt_long() in C.
128 while ((c = bc_opt_parse(&opts, bc_args_lopt)) != -1) {
134 // Barf if not allowed.
136 bc_verr(BC_ERR_FATAL_OPTION, "-e (--expression)");
138 // Add the expressions and set exit.
139 bc_args_exprs(opts.optarg);
140 vm.exit_exprs = (exit_exprs || vm.exit_exprs);
147 // Figure out if exiting on expressions is disabled.
148 if (!strcmp(opts.optarg, "-")) vm.no_exprs = true;
151 // Barf if not allowed.
153 bc_verr(BC_ERR_FATAL_OPTION, "-f (--file)");
155 // Add the expressions and set exit.
156 bc_args_file(opts.optarg);
157 vm.exit_exprs = (exit_exprs || vm.exit_exprs);
172 vm.flags |= BC_FLAG_I;
178 vm.flags |= BC_FLAG_Z;
190 vm.flags &= ~(BC_FLAG_P);
196 vm.flags &= ~(BC_FLAG_R);
204 vm.flags |= BC_FLAG_G;
211 vm.flags |= BC_FLAG_L;
218 vm.flags &= ~(BC_FLAG_Q);
224 bc_args_redefine(opts.optarg);
231 vm.flags |= BC_FLAG_S;
238 vm.flags |= BC_FLAG_W;
246 do_exit = version = true;
254 vm.flags |= DC_FLAG_X;
260 // We shouldn't get here because bc_opt_error()/bc_error() should
273 if (version) bc_vm_info(NULL);
275 vm.status = (sig_atomic_t) BC_STATUS_QUIT;
279 // We do not print the banner if expressions are used or dc is used.
280 if (!BC_IS_BC || vm.exprs.len > 1) vm.flags &= ~(BC_FLAG_Q);
282 // We need to make sure the files list is initialized. We don't want to
283 // initialize it if there are no files because it's just a waste of memory.
284 if (opts.optind < (size_t) argc && vm.files.v == NULL)
285 bc_vec_init(&vm.files, sizeof(char*), BC_DTOR_NONE);
287 // Add all the files to the vector.
288 for (i = opts.optind; i < (size_t) argc; ++i)
289 bc_vec_push(&vm.files, argv + i);