contrib/bc/src/args.c

   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 }