]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/bc/src/vm.c
MFV: xz 5.4.2.
[FreeBSD/FreeBSD.git] / contrib / bc / src / vm.c
1 /*
2  * *****************************************************************************
3  *
4  * SPDX-License-Identifier: BSD-2-Clause
5  *
6  * Copyright (c) 2018-2023 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 common to all of bc and dc.
33  *
34  */
35
36 #include <assert.h>
37 #include <ctype.h>
38 #include <errno.h>
39 #include <stdarg.h>
40 #include <string.h>
41
42 #include <signal.h>
43
44 #include <setjmp.h>
45
46 #ifndef _WIN32
47
48 #include <unistd.h>
49 #include <sys/types.h>
50 #include <unistd.h>
51
52 #else // _WIN32
53
54 #define WIN32_LEAN_AND_MEAN
55 #include <windows.h>
56 #include <io.h>
57
58 #endif // _WIN32
59
60 #include <status.h>
61 #include <vector.h>
62 #include <args.h>
63 #include <vm.h>
64 #include <read.h>
65 #include <bc.h>
66 #if BC_ENABLE_LIBRARY
67 #include <library.h>
68 #endif // BC_ENABLE_LIBRARY
69
70 #if !BC_ENABLE_LIBRARY
71
72 // The actual globals.
73 char output_bufs[BC_VM_BUF_SIZE];
74 BcVm vm_data;
75 BcVm* vm = &vm_data;
76
77 #endif // !BC_ENABLE_LIBRARY
78
79 #if BC_DEBUG_CODE
80 BC_NORETURN void
81 bc_vm_jmp(const char* f)
82 {
83 #else // BC_DEBUG_CODE
84 BC_NORETURN void
85 bc_vm_jmp(void)
86 {
87 #endif
88
89 #if BC_ENABLE_LIBRARY
90         BcVm* vm = bcl_getspecific();
91 #endif // BC_ENABLE_LIBRARY
92
93         assert(BC_SIG_EXC(vm));
94
95         BC_SIG_MAYLOCK;
96
97 #if BC_DEBUG_CODE
98         bc_file_puts(&vm->ferr, bc_flush_none, "Longjmp: ");
99         bc_file_puts(&vm->ferr, bc_flush_none, f);
100         bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
101         bc_file_flush(&vm->ferr, bc_flush_none);
102 #endif // BC_DEBUG_CODE
103
104 #if BC_DEBUG
105         assert(vm->jmp_bufs.len - (size_t) vm->sig_pop);
106 #endif // BC_DEBUG
107
108         if (vm->jmp_bufs.len == 0) abort();
109         if (vm->sig_pop) bc_vec_pop(&vm->jmp_bufs);
110         else vm->sig_pop = 1;
111
112         siglongjmp(*((sigjmp_buf*) bc_vec_top(&vm->jmp_bufs)), 1);
113 }
114
115 #if !BC_ENABLE_LIBRARY
116
117 /**
118  * Handles signals. This is the signal handler.
119  * @param sig  The signal to handle.
120  */
121 static void
122 bc_vm_sig(int sig)
123 {
124 #if BC_ENABLE_EDITLINE
125         // Editline needs this to resize the terminal. This also needs to come first
126         // because a resize always needs to happen.
127         if (sig == SIGWINCH)
128         {
129                 if (BC_TTY)
130                 {
131                         el_resize(vm->history.el);
132
133                         // If the signal was a SIGWINCH, clear it because we don't need to
134                         // print a stack trace in that case.
135                         if (vm->sig == SIGWINCH)
136                         {
137                                 vm->sig = 0;
138                         }
139                 }
140
141                 return;
142         }
143 #endif // BC_ENABLE_EDITLINE
144
145         // There is already a signal in flight if this is true.
146         if (vm->status == (sig_atomic_t) BC_STATUS_QUIT || vm->sig != 0)
147         {
148                 if (!BC_I || sig != SIGINT) vm->status = BC_STATUS_QUIT;
149                 return;
150         }
151
152         // We always want to set this because a stack trace can be printed if we do.
153         vm->sig = sig;
154
155         // Only reset under these conditions; otherwise, quit.
156         if (sig == SIGINT && BC_SIGINT && BC_I)
157         {
158                 int err = errno;
159
160 #if BC_ENABLE_EDITLINE
161                 // Editline needs this, for some unknown reason.
162                 if (write(STDOUT_FILENO, "^C", 2) != (ssize_t) 2)
163                 {
164                         vm->status = BC_STATUS_ERROR_FATAL;
165                 }
166 #endif // BC_ENABLE_EDITLINE
167
168                 // Write the message.
169                 if (write(STDOUT_FILENO, vm->sigmsg, vm->siglen) !=
170                     (ssize_t) vm->siglen)
171                 {
172                         vm->status = BC_STATUS_ERROR_FATAL;
173                 }
174
175                 errno = err;
176         }
177         else
178         {
179 #if BC_ENABLE_EDITLINE
180                 if (write(STDOUT_FILENO, "^C", 2) != (ssize_t) 2)
181                 {
182                         vm->status = BC_STATUS_ERROR_FATAL;
183                         return;
184                 }
185 #endif // BC_ENABLE_EDITLINE
186
187                 vm->status = BC_STATUS_QUIT;
188         }
189
190 #if BC_ENABLE_LINE_LIB
191         // Readline and Editline need this to actually handle sigints correctly.
192         if (sig == SIGINT && bc_history_inlinelib)
193         {
194                 bc_history_inlinelib = 0;
195                 siglongjmp(bc_history_jmpbuf, 1);
196         }
197 #endif // BC_ENABLE_LINE_LIB
198
199         assert(vm->jmp_bufs.len);
200
201         // Only jump if signals are not locked. The jump will happen by whoever
202         // unlocks signals.
203         if (!vm->sig_lock) BC_JMP;
204 }
205
206 /**
207  * Sets up signal handling.
208  */
209 static void
210 bc_vm_sigaction(void)
211 {
212 #ifndef _WIN32
213
214         struct sigaction sa;
215
216         sigemptyset(&sa.sa_mask);
217         sa.sa_flags = 0;
218
219         // This mess is to silence a warning on Clang with regards to glibc's
220         // sigaction handler, which activates the warning here.
221 #if BC_CLANG
222 #pragma clang diagnostic ignored "-Wdisabled-macro-expansion"
223 #endif // BC_CLANG
224         sa.sa_handler = bc_vm_sig;
225 #if BC_CLANG
226 #pragma clang diagnostic warning "-Wdisabled-macro-expansion"
227 #endif // BC_CLANG
228
229         sigaction(SIGTERM, &sa, NULL);
230         sigaction(SIGQUIT, &sa, NULL);
231         sigaction(SIGINT, &sa, NULL);
232
233 #if BC_ENABLE_EDITLINE
234         // Editline needs this to resize the terminal.
235         if (BC_TTY) sigaction(SIGWINCH, &sa, NULL);
236 #endif // BC_ENABLE_EDITLINE
237
238 #if BC_ENABLE_HISTORY
239         if (BC_TTY) sigaction(SIGHUP, &sa, NULL);
240 #endif // BC_ENABLE_HISTORY
241
242 #else // _WIN32
243
244         signal(SIGTERM, bc_vm_sig);
245         signal(SIGINT, bc_vm_sig);
246
247 #endif // _WIN32
248 }
249
250 void
251 bc_vm_info(const char* const help)
252 {
253         BC_SIG_ASSERT_LOCKED;
254
255         // Print the banner.
256         bc_file_printf(&vm->fout, "%s %s\n%s", vm->name, BC_VERSION, bc_copyright);
257
258         // Print the help.
259         if (help != NULL)
260         {
261                 bc_file_putchar(&vm->fout, bc_flush_none, '\n');
262
263 #if BC_ENABLED
264                 if (BC_IS_BC)
265                 {
266                         const char* const banner = BC_DEFAULT_BANNER ? "to" : "to not";
267                         const char* const sigint = BC_DEFAULT_SIGINT_RESET ? "to reset" :
268                                                                              "to exit";
269                         const char* const tty = BC_DEFAULT_TTY_MODE ? "enabled" :
270                                                                       "disabled";
271                         const char* const prompt = BC_DEFAULT_PROMPT ? "enabled" :
272                                                                        "disabled";
273                         const char* const expr = BC_DEFAULT_EXPR_EXIT ? "to exit" :
274                                                                         "to not exit";
275                         const char* const clamp = BC_DEFAULT_DIGIT_CLAMP ? "to clamp" :
276                                                                            "to not clamp";
277
278                         bc_file_printf(&vm->fout, help, vm->name, vm->name, BC_VERSION,
279                                        BC_BUILD_TYPE, banner, sigint, tty, prompt, expr,
280                                        clamp);
281                 }
282 #endif // BC_ENABLED
283
284 #if DC_ENABLED
285                 if (BC_IS_DC)
286                 {
287                         const char* const sigint = DC_DEFAULT_SIGINT_RESET ? "to reset" :
288                                                                              "to exit";
289                         const char* const tty = DC_DEFAULT_TTY_MODE ? "enabled" :
290                                                                       "disabled";
291                         const char* const prompt = DC_DEFAULT_PROMPT ? "enabled" :
292                                                                        "disabled";
293                         const char* const expr = DC_DEFAULT_EXPR_EXIT ? "to exit" :
294                                                                         "to not exit";
295                         const char* const clamp = DC_DEFAULT_DIGIT_CLAMP ? "to clamp" :
296                                                                            "to not clamp";
297
298                         bc_file_printf(&vm->fout, help, vm->name, vm->name, BC_VERSION,
299                                        BC_BUILD_TYPE, sigint, tty, prompt, expr, clamp);
300                 }
301 #endif // DC_ENABLED
302         }
303
304         // Flush.
305         bc_file_flush(&vm->fout, bc_flush_none);
306 }
307 #endif // !BC_ENABLE_LIBRARY
308
309 #if !BC_ENABLE_LIBRARY && !BC_ENABLE_MEMCHECK
310 BC_NORETURN
311 #endif // !BC_ENABLE_LIBRARY && !BC_ENABLE_MEMCHECK
312 void
313 bc_vm_fatalError(BcErr e)
314 {
315         bc_err(e);
316 #if !BC_ENABLE_LIBRARY && !BC_ENABLE_MEMCHECK
317         BC_UNREACHABLE
318 #if !BC_CLANG
319         abort();
320 #endif // !BC_CLANG
321 #endif // !BC_ENABLE_LIBRARY && !BC_ENABLE_MEMCHECK
322 }
323
324 #if BC_ENABLE_LIBRARY
325 BC_NORETURN void
326 bc_vm_handleError(BcErr e)
327 {
328 #if BC_ENABLE_LIBRARY
329         BcVm* vm = bcl_getspecific();
330 #endif // BC_ENABLE_LIBRARY
331
332         assert(e < BC_ERR_NELEMS);
333         assert(!vm->sig_pop);
334
335         BC_SIG_LOCK;
336
337         // If we have a normal error...
338         if (e <= BC_ERR_MATH_DIVIDE_BY_ZERO)
339         {
340                 // Set the error.
341                 vm->err = (BclError) (e - BC_ERR_MATH_NEGATIVE +
342                                       BCL_ERROR_MATH_NEGATIVE);
343         }
344         // Abort if we should.
345         else if (vm->abrt) abort();
346         else if (e == BC_ERR_FATAL_ALLOC_ERR) vm->err = BCL_ERROR_FATAL_ALLOC_ERR;
347         else vm->err = BCL_ERROR_FATAL_UNKNOWN_ERR;
348
349         BC_JMP;
350 }
351 #else // BC_ENABLE_LIBRARY
352 #if BC_DEBUG
353 void
354 bc_vm_handleError(BcErr e, const char* file, int fline, size_t line, ...)
355 #else // BC_DEBUG
356 void
357 bc_vm_handleError(BcErr e, size_t line, ...)
358 #endif // BC_DEBUG
359 {
360         BcStatus s;
361         va_list args;
362         uchar id = bc_err_ids[e];
363         const char* err_type = vm->err_ids[id];
364         sig_atomic_t lock;
365
366         assert(e < BC_ERR_NELEMS);
367         assert(!vm->sig_pop);
368
369 #if BC_ENABLED
370         // Figure out if the POSIX error should be an error, a warning, or nothing.
371         if (!BC_S && e >= BC_ERR_POSIX_START)
372         {
373                 if (BC_W)
374                 {
375                         // Make sure to not return an error.
376                         id = UCHAR_MAX;
377                         err_type = vm->err_ids[BC_ERR_IDX_WARN];
378                 }
379                 else return;
380         }
381 #endif // BC_ENABLED
382
383         BC_SIG_TRYLOCK(lock);
384
385         // Make sure all of stdout is written first.
386         s = bc_file_flushErr(&vm->fout, bc_flush_err);
387
388         // Just jump out if the flush failed; there's nothing we can do.
389         if (BC_ERR(s == BC_STATUS_ERROR_FATAL))
390         {
391                 vm->status = (sig_atomic_t) s;
392                 BC_JMP;
393         }
394
395         // Print the error message.
396         va_start(args, line);
397         bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
398         bc_file_puts(&vm->ferr, bc_flush_none, err_type);
399         bc_file_putchar(&vm->ferr, bc_flush_none, ' ');
400         bc_file_vprintf(&vm->ferr, vm->err_msgs[e], args);
401         va_end(args);
402
403         // Print the extra information if we have it.
404         if (BC_NO_ERR(vm->file != NULL))
405         {
406                 // This is the condition for parsing vs runtime.
407                 // If line is not 0, it is parsing.
408                 if (line)
409                 {
410                         bc_file_puts(&vm->ferr, bc_flush_none, "\n    ");
411                         bc_file_puts(&vm->ferr, bc_flush_none, vm->file);
412                         bc_file_printf(&vm->ferr, ":%zu\n", line);
413                 }
414                 else
415                 {
416                         // Print a stack trace.
417                         bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
418                         bc_program_printStackTrace(&vm->prog);
419                 }
420         }
421         else
422         {
423                 bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
424         }
425
426 #if BC_DEBUG
427         bc_file_printf(&vm->ferr, "\n    %s:%d\n", file, fline);
428 #endif // BC_DEBUG
429
430         bc_file_puts(&vm->ferr, bc_flush_none, "\n");
431
432         s = bc_file_flushErr(&vm->ferr, bc_flush_err);
433
434 #if !BC_ENABLE_MEMCHECK
435         // Because this function is called by a BC_NORETURN function when fatal
436         // errors happen, we need to make sure to exit on fatal errors. This will
437         // be faster anyway. This function *cannot jump when a fatal error occurs!*
438         if (BC_ERR(id == BC_ERR_IDX_FATAL || s == BC_STATUS_ERROR_FATAL))
439         {
440                 exit(bc_vm_atexit((int) BC_STATUS_ERROR_FATAL));
441         }
442 #else // !BC_ENABLE_MEMCHECK
443         if (BC_ERR(s == BC_STATUS_ERROR_FATAL)) vm->status = (sig_atomic_t) s;
444         else
445 #endif // !BC_ENABLE_MEMCHECK
446         {
447                 vm->status = (sig_atomic_t) (uchar) (id + 1);
448         }
449
450         // Only jump if there is an error.
451         if (BC_ERR(vm->status)) BC_JMP;
452
453         BC_SIG_TRYUNLOCK(lock);
454 }
455
456 char*
457 bc_vm_getenv(const char* var)
458 {
459         char* ret;
460
461 #ifndef _WIN32
462         ret = getenv(var);
463 #else // _WIN32
464         _dupenv_s(&ret, NULL, var);
465 #endif // _WIN32
466
467         return ret;
468 }
469
470 void
471 bc_vm_getenvFree(char* val)
472 {
473         BC_UNUSED(val);
474 #ifdef _WIN32
475         free(val);
476 #endif // _WIN32
477 }
478
479 /**
480  * Sets a flag from an environment variable and the default.
481  * @param var   The environment variable.
482  * @param def   The default.
483  * @param flag  The flag to set.
484  */
485 static void
486 bc_vm_setenvFlag(const char* const var, int def, uint16_t flag)
487 {
488         // Get the value.
489         char* val = bc_vm_getenv(var);
490
491         // If there is no value...
492         if (val == NULL)
493         {
494                 // Set the default.
495                 if (def) vm->flags |= flag;
496                 else vm->flags &= ~(flag);
497         }
498         // Parse the value.
499         else if (strtoul(val, NULL, 0)) vm->flags |= flag;
500         else vm->flags &= ~(flag);
501
502         bc_vm_getenvFree(val);
503 }
504
505 /**
506  * Parses the arguments in {B,D]C_ENV_ARGS.
507  * @param env_args_name  The environment variable to use.
508  * @param scale          A pointer to return the scale that the arguments set,
509  *                       if any.
510  * @param ibase          A pointer to return the ibase that the arguments set,
511  *                       if any.
512  * @param obase          A pointer to return the obase that the arguments set,
513  *                       if any.
514  */
515 static void
516 bc_vm_envArgs(const char* const env_args_name, BcBigDig* scale, BcBigDig* ibase,
517               BcBigDig* obase)
518 {
519         char *env_args = bc_vm_getenv(env_args_name), *buf, *start;
520         char instr = '\0';
521
522         BC_SIG_ASSERT_LOCKED;
523
524         if (env_args == NULL) return;
525
526                 // Windows already allocates, so we don't need to.
527 #ifndef _WIN32
528         start = buf = vm->env_args_buffer = bc_vm_strdup(env_args);
529 #else // _WIN32
530         start = buf = vm->env_args_buffer = env_args;
531 #endif // _WIN32
532
533         assert(buf != NULL);
534
535         // Create two buffers for parsing. These need to stay throughout the entire
536         // execution of bc, unfortunately, because of filenames that might be in
537         // there.
538         bc_vec_init(&vm->env_args, sizeof(char*), BC_DTOR_NONE);
539         bc_vec_push(&vm->env_args, &env_args_name);
540
541         // While we haven't reached the end of the args...
542         while (*buf)
543         {
544                 // If we don't have whitespace...
545                 if (!isspace(*buf))
546                 {
547                         // If we have the start of a string...
548                         if (*buf == '"' || *buf == '\'')
549                         {
550                                 // Set stuff appropriately.
551                                 instr = *buf;
552                                 buf += 1;
553
554                                 // Check for the empty string.
555                                 if (*buf == instr)
556                                 {
557                                         instr = '\0';
558                                         buf += 1;
559                                         continue;
560                                 }
561                         }
562
563                         // Push the pointer to the args buffer.
564                         bc_vec_push(&vm->env_args, &buf);
565
566                         // Parse the string.
567                         while (*buf &&
568                                ((!instr && !isspace(*buf)) || (instr && *buf != instr)))
569                         {
570                                 buf += 1;
571                         }
572
573                         // If we did find the end of the string...
574                         if (*buf)
575                         {
576                                 if (instr) instr = '\0';
577
578                                 // Reset stuff.
579                                 *buf = '\0';
580                                 buf += 1;
581                                 start = buf;
582                         }
583                         else if (instr) bc_error(BC_ERR_FATAL_OPTION, 0, start);
584                 }
585                 // If we have whitespace, eat it.
586                 else buf += 1;
587         }
588
589         // Make sure to push a NULL pointer at the end.
590         buf = NULL;
591         bc_vec_push(&vm->env_args, &buf);
592
593         // Parse the arguments.
594         bc_args((int) vm->env_args.len - 1, bc_vec_item(&vm->env_args, 0), false,
595                 scale, ibase, obase);
596 }
597
598 /**
599  * Gets the {B,D}C_LINE_LENGTH.
600  * @param var  The environment variable to pull it from.
601  * @return     The line length.
602  */
603 static size_t
604 bc_vm_envLen(const char* var)
605 {
606         char* lenv = bc_vm_getenv(var);
607         size_t i, len = BC_NUM_PRINT_WIDTH;
608         int num;
609
610         // Return the default with none.
611         if (lenv == NULL) return len;
612
613         len = strlen(lenv);
614
615         // Figure out if it's a number.
616         for (num = 1, i = 0; num && i < len; ++i)
617         {
618                 num = isdigit(lenv[i]);
619         }
620
621         // If it is a number...
622         if (num)
623         {
624                 // Parse it and clamp it if needed.
625                 len = (size_t) atoi(lenv) - 1;
626                 if (len == 1 || len >= UINT16_MAX) len = BC_NUM_PRINT_WIDTH;
627         }
628         // Set the default.
629         else len = BC_NUM_PRINT_WIDTH;
630
631         bc_vm_getenvFree(lenv);
632
633         return len;
634 }
635 #endif // BC_ENABLE_LIBRARY
636
637 void
638 bc_vm_shutdown(void)
639 {
640         BC_SIG_ASSERT_LOCKED;
641
642 #if BC_ENABLE_NLS
643         if (vm->catalog != BC_VM_INVALID_CATALOG) catclose(vm->catalog);
644 #endif // BC_ENABLE_NLS
645
646 #if !BC_ENABLE_LIBRARY
647 #if BC_ENABLE_HISTORY
648         // This must always run to ensure that the terminal is back to normal, i.e.,
649         // has raw mode disabled. But we should only do it if we did not have a bad
650         // terminal because history was not initialized if it is a bad terminal.
651         if (BC_TTY && !vm->history.badTerm) bc_history_free(&vm->history);
652 #endif // BC_ENABLE_HISTORY
653 #endif // !BC_ENABLE_LIBRARY
654
655 #if BC_DEBUG
656 #if !BC_ENABLE_LIBRARY
657         bc_vec_free(&vm->env_args);
658         free(vm->env_args_buffer);
659         bc_vec_free(&vm->files);
660         bc_vec_free(&vm->exprs);
661
662         if (BC_PARSE_IS_INITED(&vm->read_prs, &vm->prog))
663         {
664                 bc_vec_free(&vm->read_buf);
665                 bc_parse_free(&vm->read_prs);
666         }
667
668         bc_parse_free(&vm->prs);
669         bc_program_free(&vm->prog);
670
671         bc_slabvec_free(&vm->slabs);
672 #endif // !BC_ENABLE_LIBRARY
673
674         bc_vm_freeTemps();
675 #endif // BC_DEBUG
676
677 #if !BC_ENABLE_LIBRARY
678         // We always want to flush.
679         bc_file_free(&vm->fout);
680         bc_file_free(&vm->ferr);
681 #endif // !BC_ENABLE_LIBRARY
682 }
683
684 void
685 bc_vm_addTemp(BcDig* num)
686 {
687 #if BC_ENABLE_LIBRARY
688         BcVm* vm = bcl_getspecific();
689 #endif // BC_ENABLE_LIBRARY
690
691         BC_SIG_ASSERT_LOCKED;
692
693         // If we don't have room, just free.
694         if (vm->temps_len == BC_VM_MAX_TEMPS) free(num);
695         else
696         {
697                 // Add to the buffer and length.
698                 vm->temps_buf[vm->temps_len] = num;
699                 vm->temps_len += 1;
700         }
701 }
702
703 BcDig*
704 bc_vm_takeTemp(void)
705 {
706 #if BC_ENABLE_LIBRARY
707         BcVm* vm = bcl_getspecific();
708 #endif // BC_ENABLE_LIBRARY
709
710         BC_SIG_ASSERT_LOCKED;
711
712         if (!vm->temps_len) return NULL;
713
714         vm->temps_len -= 1;
715
716         return vm->temps_buf[vm->temps_len];
717 }
718
719 BcDig*
720 bc_vm_getTemp(void)
721 {
722 #if BC_ENABLE_LIBRARY
723         BcVm* vm = bcl_getspecific();
724 #endif // BC_ENABLE_LIBRARY
725
726         BC_SIG_ASSERT_LOCKED;
727
728         if (!vm->temps_len) return NULL;
729
730         return vm->temps_buf[vm->temps_len - 1];
731 }
732
733 void
734 bc_vm_freeTemps(void)
735 {
736         size_t i;
737 #if BC_ENABLE_LIBRARY
738         BcVm* vm = bcl_getspecific();
739 #endif // BC_ENABLE_LIBRARY
740
741         BC_SIG_ASSERT_LOCKED;
742
743         if (!vm->temps_len) return;
744
745         // Free them all...
746         for (i = 0; i < vm->temps_len; ++i)
747         {
748                 free(vm->temps_buf[i]);
749         }
750
751         vm->temps_len = 0;
752 }
753
754 #if !BC_ENABLE_LIBRARY
755
756 size_t
757 bc_vm_numDigits(size_t val)
758 {
759         size_t digits = 0;
760
761         do
762         {
763                 digits += 1;
764                 val /= 10;
765         }
766         while (val != 0);
767
768         return digits;
769 }
770
771 #endif // !BC_ENABLE_LIBRARY
772
773 inline size_t
774 bc_vm_arraySize(size_t n, size_t size)
775 {
776         size_t res = n * size;
777
778         if (BC_ERR(BC_VM_MUL_OVERFLOW(n, size, res)))
779         {
780                 bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
781         }
782
783         return res;
784 }
785
786 inline size_t
787 bc_vm_growSize(size_t a, size_t b)
788 {
789         size_t res = a + b;
790
791         if (BC_ERR(res >= SIZE_MAX || res < a))
792         {
793                 bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
794         }
795
796         return res;
797 }
798
799 void*
800 bc_vm_malloc(size_t n)
801 {
802         void* ptr;
803
804         BC_SIG_ASSERT_LOCKED;
805
806         ptr = malloc(n);
807
808         if (BC_ERR(ptr == NULL))
809         {
810                 bc_vm_freeTemps();
811
812                 ptr = malloc(n);
813
814                 if (BC_ERR(ptr == NULL)) bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
815         }
816
817         return ptr;
818 }
819
820 void*
821 bc_vm_realloc(void* ptr, size_t n)
822 {
823         void* temp;
824
825         BC_SIG_ASSERT_LOCKED;
826
827         temp = realloc(ptr, n);
828
829         if (BC_ERR(temp == NULL))
830         {
831                 bc_vm_freeTemps();
832
833                 temp = realloc(ptr, n);
834
835                 if (BC_ERR(temp == NULL)) bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
836         }
837
838         return temp;
839 }
840
841 char*
842 bc_vm_strdup(const char* str)
843 {
844         char* s;
845
846         BC_SIG_ASSERT_LOCKED;
847
848         s = strdup(str);
849
850         if (BC_ERR(s == NULL))
851         {
852                 bc_vm_freeTemps();
853
854                 s = strdup(str);
855
856                 if (BC_ERR(s == NULL)) bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
857         }
858
859         return s;
860 }
861
862 #if !BC_ENABLE_LIBRARY
863 void
864 bc_vm_printf(const char* fmt, ...)
865 {
866         va_list args;
867 #if BC_ENABLE_LIBRARY
868         BcVm* vm = bcl_getspecific();
869 #else // BC_ENABLE_LIBRARY
870         sig_atomic_t lock;
871 #endif // BC_ENABLE_LIBRARY
872
873         BC_SIG_TRYLOCK(lock);
874
875         va_start(args, fmt);
876         bc_file_vprintf(&vm->fout, fmt, args);
877         va_end(args);
878
879         vm->nchars = 0;
880
881         BC_SIG_TRYUNLOCK(lock);
882 }
883 #endif // !BC_ENABLE_LIBRARY
884
885 void
886 bc_vm_putchar(int c, BcFlushType type)
887 {
888 #if BC_ENABLE_LIBRARY
889         BcVm* vm = bcl_getspecific();
890         bc_vec_pushByte(&vm->out, (uchar) c);
891 #else // BC_ENABLE_LIBRARY
892         bc_file_putchar(&vm->fout, type, (uchar) c);
893         vm->nchars = (c == '\n' ? 0 : vm->nchars + 1);
894 #endif // BC_ENABLE_LIBRARY
895 }
896
897 #if !BC_ENABLE_LIBRARY
898
899 #ifdef __OpenBSD__
900
901 /**
902  * Aborts with a message. This should never be called because I have carefully
903  * made sure that the calls to pledge() and unveil() are correct, but it's here
904  * just in case.
905  * @param msg  The message to print.
906  */
907 BC_NORETURN static void
908 bc_abortm(const char* msg)
909 {
910         bc_file_puts(&vm->ferr, bc_flush_none, msg);
911         bc_file_puts(&vm->ferr, bc_flush_none, "; this is a bug");
912         bc_file_flush(&vm->ferr, bc_flush_none);
913         abort();
914 }
915
916 void
917 bc_pledge(const char* promises, const char* execpromises)
918 {
919         int r = pledge(promises, execpromises);
920         if (r) bc_abortm("pledge() failed");
921 }
922
923 #if BC_ENABLE_EXTRA_MATH
924
925 /**
926  * A convenience and portability function for OpenBSD's unveil().
927  * @param path         The path.
928  * @param permissions  The permissions for the path.
929  */
930 static void
931 bc_unveil(const char* path, const char* permissions)
932 {
933         int r = unveil(path, permissions);
934         if (r) bc_abortm("unveil() failed");
935 }
936
937 #endif // BC_ENABLE_EXTRA_MATH
938
939 #else // __OpenBSD__
940
941 void
942 bc_pledge(const char* promises, const char* execpromises)
943 {
944         BC_UNUSED(promises);
945         BC_UNUSED(execpromises);
946 }
947
948 #if BC_ENABLE_EXTRA_MATH
949 static void
950 bc_unveil(const char* path, const char* permissions)
951 {
952         BC_UNUSED(path);
953         BC_UNUSED(permissions);
954 }
955 #endif // BC_ENABLE_EXTRA_MATH
956
957 #endif // __OpenBSD__
958
959 /**
960  * Cleans unneeded variables, arrays, functions, strings, and constants when
961  * done executing a line of stdin. This is to prevent memory usage growing
962  * without bound. This is an idea from busybox.
963  */
964 static void
965 bc_vm_clean(void)
966 {
967         BcVec* fns = &vm->prog.fns;
968         BcFunc* f = bc_vec_item(fns, BC_PROG_MAIN);
969         BcInstPtr* ip = bc_vec_item(&vm->prog.stack, 0);
970         bool good = ((vm->status && vm->status != BC_STATUS_QUIT) || vm->sig != 0);
971
972         BC_SIG_ASSERT_LOCKED;
973
974         // If all is good, go ahead and reset.
975         if (good) bc_program_reset(&vm->prog);
976
977 #if BC_ENABLED
978         // bc has this extra condition. If it not satisfied, it is in the middle of
979         // a parse.
980         if (good && BC_IS_BC) good = !BC_PARSE_NO_EXEC(&vm->prs);
981 #endif // BC_ENABLED
982
983 #if DC_ENABLED
984         // For dc, it is safe only when all of the results on the results stack are
985         // safe, which means that they are temporaries or other things that don't
986         // need strings or constants.
987         if (BC_IS_DC)
988         {
989                 size_t i;
990
991                 good = true;
992
993                 for (i = 0; good && i < vm->prog.results.len; ++i)
994                 {
995                         BcResult* r = (BcResult*) bc_vec_item(&vm->prog.results, i);
996                         good = BC_VM_SAFE_RESULT(r);
997                 }
998         }
999 #endif // DC_ENABLED
1000
1001         // If this condition is true, we can get rid of strings,
1002         // constants, and code.
1003         if (good && vm->prog.stack.len == 1 && ip->idx == f->code.len)
1004         {
1005                 // XXX: Nothing can be popped in dc. Deal with it.
1006
1007 #if BC_ENABLED
1008                 if (BC_IS_BC)
1009                 {
1010                         // XXX: you cannot delete strings, functions, or constants in bc.
1011                         // Deal with it.
1012                         bc_vec_popAll(&f->labels);
1013                 }
1014 #endif // BC_ENABLED
1015
1016                 bc_vec_popAll(&f->code);
1017
1018                 ip->idx = 0;
1019         }
1020 }
1021
1022 /**
1023  * Process a bunch of text.
1024  * @param text  The text to process.
1025  * @param mode  The mode to process in.
1026  */
1027 static void
1028 bc_vm_process(const char* text, BcMode mode)
1029 {
1030         // Set up the parser.
1031         bc_parse_text(&vm->prs, text, mode);
1032
1033         while (vm->prs.l.t != BC_LEX_EOF)
1034         {
1035                 // Parsing requires a signal lock. We also don't parse everything; we
1036                 // want to execute as soon as possible for *everything*.
1037                 BC_SIG_LOCK;
1038                 vm->parse(&vm->prs);
1039                 BC_SIG_UNLOCK;
1040
1041                 // Execute if possible.
1042                 if (BC_IS_DC || !BC_PARSE_NO_EXEC(&vm->prs)) bc_program_exec(&vm->prog);
1043
1044                 assert(BC_IS_DC || vm->prog.results.len == 0);
1045
1046                 // Flush in interactive mode.
1047                 if (BC_I) bc_file_flush(&vm->fout, bc_flush_save);
1048         }
1049 }
1050
1051 #if BC_ENABLED
1052
1053 /**
1054  * Ends a series of if statements. This is to ensure that full parses happen
1055  * when a file finishes or stdin has no more data. Without this, bc thinks that
1056  * it cannot parse any further. But if we reach the end of a file or stdin has
1057  * no more data, we know we can add an empty else clause.
1058  */
1059 static void
1060 bc_vm_endif(void)
1061 {
1062         bc_parse_endif(&vm->prs);
1063         bc_program_exec(&vm->prog);
1064 }
1065
1066 #endif // BC_ENABLED
1067
1068 /**
1069  * Processes a file.
1070  * @param file  The filename.
1071  */
1072 static void
1073 bc_vm_file(const char* file)
1074 {
1075         char* data = NULL;
1076 #if BC_ENABLE_LIBRARY
1077         BcVm* vm = bcl_getspecific();
1078 #endif // BC_ENABLE_LIBRARY
1079
1080         assert(!vm->sig_pop);
1081
1082         vm->mode = BC_MODE_FILE;
1083
1084         // Set up the lexer.
1085         bc_lex_file(&vm->prs.l, file);
1086
1087         BC_SIG_LOCK;
1088
1089         // Read the file.
1090         data = bc_read_file(file);
1091
1092         assert(data != NULL);
1093
1094         BC_SETJMP_LOCKED(vm, err);
1095
1096         BC_SIG_UNLOCK;
1097
1098         // Process it.
1099         bc_vm_process(data, BC_MODE_FILE);
1100
1101 #if BC_ENABLED
1102         // Make sure to end any open if statements.
1103         if (BC_IS_BC) bc_vm_endif();
1104 #endif // BC_ENABLED
1105
1106 err:
1107
1108         BC_SIG_MAYLOCK;
1109
1110         // Cleanup.
1111         free(data);
1112         bc_vm_clean();
1113
1114         // bc_program_reset(), called by bc_vm_clean(), resets the status.
1115         // We want it to clear the sig_pop variable in case it was set.
1116         if (vm->status == (sig_atomic_t) BC_STATUS_SUCCESS) BC_LONGJMP_STOP;
1117
1118         BC_LONGJMP_CONT(vm);
1119 }
1120
1121 bool
1122 bc_vm_readLine(bool clear)
1123 {
1124         BcStatus s;
1125         bool good;
1126
1127         BC_SIG_ASSERT_NOT_LOCKED;
1128
1129         // Clear the buffer if desired.
1130         if (clear) bc_vec_empty(&vm->buffer);
1131
1132         // Empty the line buffer.
1133         bc_vec_empty(&vm->line_buf);
1134
1135         if (vm->eof) return false;
1136
1137         do
1138         {
1139                 // bc_read_line() must always return either BC_STATUS_SUCCESS or
1140                 // BC_STATUS_EOF. Everything else, it and whatever it calls, must jump
1141                 // out instead.
1142                 s = bc_read_line(&vm->line_buf, ">>> ");
1143                 vm->eof = (s == BC_STATUS_EOF);
1144         }
1145         while (s == BC_STATUS_SUCCESS && !vm->eof && vm->line_buf.len < 1);
1146
1147         good = (vm->line_buf.len > 1);
1148
1149         // Concat if we found something.
1150         if (good) bc_vec_concat(&vm->buffer, vm->line_buf.v);
1151
1152         return good;
1153 }
1154
1155 /**
1156  * Processes text from stdin.
1157  */
1158 static void
1159 bc_vm_stdin(void)
1160 {
1161         bool clear;
1162
1163 #if BC_ENABLE_LIBRARY
1164         BcVm* vm = bcl_getspecific();
1165 #endif // BC_ENABLE_LIBRARY
1166
1167         clear = true;
1168         vm->mode = BC_MODE_STDIN;
1169
1170         // Set up the lexer.
1171         bc_lex_file(&vm->prs.l, bc_program_stdin_name);
1172
1173         // These are global so that the lexers can access them, but they are
1174         // allocated and freed in this function because they should only be used for
1175         // stdin and expressions (they are used in bc_vm_exprs() as well). So they
1176         // are tied to this function, really. Well, this and bc_vm_readLine(). These
1177         // are the reasons that we have vm->is_stdin to tell the lexers if we are
1178         // reading from stdin. Well, both lexers care. And the reason they care is
1179         // so that if a comment or a string goes across multiple lines, the lexer
1180         // can request more data from stdin until the comment or string is ended.
1181         BC_SIG_LOCK;
1182         bc_vec_init(&vm->buffer, sizeof(uchar), BC_DTOR_NONE);
1183         bc_vec_init(&vm->line_buf, sizeof(uchar), BC_DTOR_NONE);
1184         BC_SETJMP_LOCKED(vm, err);
1185         BC_SIG_UNLOCK;
1186
1187 // This label exists because errors can cause jumps to end up at the err label
1188 // below. If that happens, and the error should be cleared and execution
1189 // continue, then we need to jump back.
1190 restart:
1191
1192         // While we still read data from stdin.
1193         while (bc_vm_readLine(clear))
1194         {
1195                 size_t len = vm->buffer.len - 1;
1196                 const char* str = vm->buffer.v;
1197
1198                 // We don't want to clear the buffer when the line ends with a backslash
1199                 // because a backslash newline is special in bc.
1200                 clear = (len < 2 || str[len - 2] != '\\' || str[len - 1] != '\n');
1201                 if (!clear) continue;
1202
1203                 // Process the data.
1204                 bc_vm_process(vm->buffer.v, BC_MODE_STDIN);
1205
1206                 if (vm->eof) break;
1207                 else
1208                 {
1209                         BC_SIG_LOCK;
1210                         bc_vm_clean();
1211                         BC_SIG_UNLOCK;
1212                 }
1213         }
1214
1215 #if BC_ENABLED
1216         // End the if statements.
1217         if (BC_IS_BC) bc_vm_endif();
1218 #endif // BC_ENABLED
1219
1220 err:
1221
1222         BC_SIG_MAYLOCK;
1223
1224         // Cleanup.
1225         bc_vm_clean();
1226
1227 #if !BC_ENABLE_MEMCHECK
1228         assert(vm->status != BC_STATUS_ERROR_FATAL);
1229
1230         vm->status = vm->status == BC_STATUS_QUIT || !BC_I ? vm->status :
1231                                                              BC_STATUS_SUCCESS;
1232 #else // !BC_ENABLE_MEMCHECK
1233         vm->status = vm->status == BC_STATUS_ERROR_FATAL ||
1234                              vm->status == BC_STATUS_QUIT || !BC_I ?
1235                          vm->status :
1236                          BC_STATUS_SUCCESS;
1237 #endif // !BC_ENABLE_MEMCHECK
1238
1239         if (!vm->status && !vm->eof)
1240         {
1241                 bc_vec_empty(&vm->buffer);
1242                 BC_LONGJMP_STOP;
1243                 BC_SIG_UNLOCK;
1244                 goto restart;
1245         }
1246
1247 #if BC_DEBUG
1248         // Since these are tied to this function, free them here. We only free in
1249         // debug mode because stdin is always the last thing read.
1250         bc_vec_free(&vm->line_buf);
1251         bc_vec_free(&vm->buffer);
1252 #endif // BC_DEBUG
1253
1254         BC_LONGJMP_CONT(vm);
1255 }
1256
1257 bool
1258 bc_vm_readBuf(bool clear)
1259 {
1260         size_t len = vm->exprs.len - 1;
1261         bool more;
1262
1263         BC_SIG_ASSERT_NOT_LOCKED;
1264
1265         // Clear the buffer if desired.
1266         if (clear) bc_vec_empty(&vm->buffer);
1267
1268         // We want to pop the nul byte off because that's what bc_read_buf()
1269         // expects.
1270         bc_vec_pop(&vm->buffer);
1271
1272         // Read one line of expressions.
1273         more = bc_read_buf(&vm->buffer, vm->exprs.v, &len);
1274         bc_vec_pushByte(&vm->buffer, '\0');
1275
1276         return more;
1277 }
1278
1279 static void
1280 bc_vm_exprs(void)
1281 {
1282         bool clear;
1283
1284 #if BC_ENABLE_LIBRARY
1285         BcVm* vm = bcl_getspecific();
1286 #endif // BC_ENABLE_LIBRARY
1287
1288         clear = true;
1289         vm->mode = BC_MODE_EXPRS;
1290
1291         // Prepare the lexer.
1292         bc_lex_file(&vm->prs.l, bc_program_exprs_name);
1293
1294         // We initialize this so that the lexer can access it in the case that it
1295         // needs more data for expressions, such as for a multiline string or
1296         // comment. See the comment on the allocation of vm->buffer above in
1297         // bc_vm_stdin() for more information.
1298         BC_SIG_LOCK;
1299         bc_vec_init(&vm->buffer, sizeof(uchar), BC_DTOR_NONE);
1300         BC_SETJMP_LOCKED(vm, err);
1301         BC_SIG_UNLOCK;
1302
1303         while (bc_vm_readBuf(clear))
1304         {
1305                 size_t len = vm->buffer.len - 1;
1306                 const char* str = vm->buffer.v;
1307
1308                 // We don't want to clear the buffer when the line ends with a backslash
1309                 // because a backslash newline is special in bc.
1310                 clear = (len < 2 || str[len - 2] != '\\' || str[len - 1] != '\n');
1311                 if (!clear) continue;
1312
1313                 // Process the data.
1314                 bc_vm_process(vm->buffer.v, BC_MODE_EXPRS);
1315         }
1316
1317         // If we were not supposed to clear, then we should process everything. This
1318         // makes sure that errors get reported.
1319         if (!clear) bc_vm_process(vm->buffer.v, BC_MODE_EXPRS);
1320
1321 err:
1322
1323         BC_SIG_MAYLOCK;
1324
1325         // Cleanup.
1326         bc_vm_clean();
1327
1328         // bc_program_reset(), called by bc_vm_clean(), resets the status.
1329         // We want it to clear the sig_pop variable in case it was set.
1330         if (vm->status == (sig_atomic_t) BC_STATUS_SUCCESS) BC_LONGJMP_STOP;
1331
1332         // Since this is tied to this function, free it here. We always free it here
1333         // because bc_vm_stdin() may or may not use it later.
1334         bc_vec_free(&vm->buffer);
1335
1336         BC_LONGJMP_CONT(vm);
1337 }
1338
1339 #if BC_ENABLED
1340
1341 /**
1342  * Loads a math library.
1343  * @param name  The name of the library.
1344  * @param text  The text of the source code.
1345  */
1346 static void
1347 bc_vm_load(const char* name, const char* text)
1348 {
1349         bc_lex_file(&vm->prs.l, name);
1350         bc_parse_text(&vm->prs, text, BC_MODE_FILE);
1351
1352         BC_SIG_LOCK;
1353
1354         while (vm->prs.l.t != BC_LEX_EOF)
1355         {
1356                 vm->parse(&vm->prs);
1357         }
1358
1359         BC_SIG_UNLOCK;
1360 }
1361
1362 #endif // BC_ENABLED
1363
1364 /**
1365  * Loads the default error messages.
1366  */
1367 static void
1368 bc_vm_defaultMsgs(void)
1369 {
1370         size_t i;
1371
1372         // Load the error categories.
1373         for (i = 0; i < BC_ERR_IDX_NELEMS + BC_ENABLED; ++i)
1374         {
1375                 vm->err_ids[i] = bc_errs[i];
1376         }
1377
1378         // Load the error messages.
1379         for (i = 0; i < BC_ERR_NELEMS; ++i)
1380         {
1381                 vm->err_msgs[i] = bc_err_msgs[i];
1382         }
1383 }
1384
1385 /**
1386  * Loads the error messages for the locale. If NLS is disabled, this just loads
1387  * the default messages.
1388  */
1389 static void
1390 bc_vm_gettext(void)
1391 {
1392 #if BC_ENABLE_NLS
1393         uchar id = 0;
1394         int set, msg = 1;
1395         size_t i;
1396
1397         // If no locale, load the defaults.
1398         if (vm->locale == NULL)
1399         {
1400                 vm->catalog = BC_VM_INVALID_CATALOG;
1401                 bc_vm_defaultMsgs();
1402                 return;
1403         }
1404
1405         vm->catalog = catopen(BC_MAINEXEC, NL_CAT_LOCALE);
1406
1407         // If no catalog, load the defaults.
1408         if (vm->catalog == BC_VM_INVALID_CATALOG)
1409         {
1410                 bc_vm_defaultMsgs();
1411                 return;
1412         }
1413
1414         // Load the error categories.
1415         for (set = 1; msg <= BC_ERR_IDX_NELEMS + BC_ENABLED; ++msg)
1416         {
1417                 vm->err_ids[msg - 1] = catgets(vm->catalog, set, msg, bc_errs[msg - 1]);
1418         }
1419
1420         i = 0;
1421         id = bc_err_ids[i];
1422
1423         // Load the error messages. In order to understand this loop, you must know
1424         // the order of messages and categories in the enum and in the locale files.
1425         for (set = id + 2, msg = 1; i < BC_ERR_NELEMS; ++i, ++msg)
1426         {
1427                 if (id != bc_err_ids[i])
1428                 {
1429                         msg = 1;
1430                         id = bc_err_ids[i];
1431                         set = id + 2;
1432                 }
1433
1434                 vm->err_msgs[i] = catgets(vm->catalog, set, msg, bc_err_msgs[i]);
1435         }
1436 #else // BC_ENABLE_NLS
1437         bc_vm_defaultMsgs();
1438 #endif // BC_ENABLE_NLS
1439 }
1440
1441 /**
1442  * Starts execution. Really, this is a function of historical accident; it could
1443  * probably be combined with bc_vm_boot(), but I don't care enough. Really, this
1444  * function starts when execution of bc or dc source code starts.
1445  */
1446 static void
1447 bc_vm_exec(void)
1448 {
1449         size_t i;
1450 #if DC_ENABLED
1451         bool has_file = false;
1452 #endif // DC_ENABLED
1453
1454 #if BC_ENABLED
1455         // Load the math libraries.
1456         if (BC_IS_BC && (vm->flags & BC_FLAG_L))
1457         {
1458                 // Can't allow redefinitions in the builtin library.
1459                 vm->no_redefine = true;
1460
1461                 bc_vm_load(bc_lib_name, bc_lib);
1462
1463 #if BC_ENABLE_EXTRA_MATH
1464                 if (!BC_IS_POSIX) bc_vm_load(bc_lib2_name, bc_lib2);
1465 #endif // BC_ENABLE_EXTRA_MATH
1466
1467                 // Make sure to clear this.
1468                 vm->no_redefine = false;
1469
1470                 // Execute to ensure that all is hunky dory. Without this, scale can be
1471                 // set improperly.
1472                 bc_program_exec(&vm->prog);
1473         }
1474 #endif // BC_ENABLED
1475
1476         // If there are expressions to execute...
1477         if (vm->exprs.len)
1478         {
1479                 // Process the expressions.
1480                 bc_vm_exprs();
1481
1482                 // Sometimes, executing expressions means we need to quit.
1483                 if (!vm->no_exprs && vm->exit_exprs && BC_EXPR_EXIT) return;
1484         }
1485
1486         // Process files.
1487         for (i = 0; i < vm->files.len; ++i)
1488         {
1489                 char* path = *((char**) bc_vec_item(&vm->files, i));
1490                 if (!strcmp(path, "")) continue;
1491 #if DC_ENABLED
1492                 has_file = true;
1493 #endif // DC_ENABLED
1494                 bc_vm_file(path);
1495         }
1496
1497 #if BC_ENABLE_EXTRA_MATH
1498         // These are needed for the pseudo-random number generator.
1499         bc_unveil("/dev/urandom", "r");
1500         bc_unveil("/dev/random", "r");
1501         bc_unveil(NULL, NULL);
1502 #endif // BC_ENABLE_EXTRA_MATH
1503
1504 #if BC_ENABLE_HISTORY
1505
1506         // We need to keep tty if history is enabled, and we need to keep rpath for
1507         // the times when we read from /dev/urandom.
1508         if (BC_TTY && !vm->history.badTerm) bc_pledge(bc_pledge_end_history, NULL);
1509         else
1510 #endif // BC_ENABLE_HISTORY
1511         {
1512                 bc_pledge(bc_pledge_end, NULL);
1513         }
1514
1515 #if BC_ENABLE_AFL
1516         // This is the thing that makes fuzzing with AFL++ so fast. If you move this
1517         // back, you won't cause any problems, but fuzzing will slow down. If you
1518         // move this forward, you won't fuzz anything because you will be skipping
1519         // the reading from stdin.
1520         __AFL_INIT();
1521 #endif // BC_ENABLE_AFL
1522
1523         // Execute from stdin. bc always does.
1524         if (BC_VM_RUN_STDIN(has_file)) bc_vm_stdin();
1525 }
1526
1527 void
1528 bc_vm_boot(int argc, char* argv[])
1529 {
1530         int ttyin, ttyout, ttyerr;
1531         bool tty;
1532         const char* const env_len = BC_VM_LINE_LENGTH_STR;
1533         const char* const env_args = BC_VM_ENV_ARGS_STR;
1534         const char* const env_exit = BC_VM_EXPR_EXIT_STR;
1535         const char* const env_clamp = BC_VM_DIGIT_CLAMP_STR;
1536         int env_exit_def = BC_VM_EXPR_EXIT_DEF;
1537         int env_clamp_def = BC_VM_DIGIT_CLAMP_DEF;
1538         BcBigDig scale = BC_NUM_BIGDIG_MAX;
1539         BcBigDig env_scale = BC_NUM_BIGDIG_MAX;
1540         BcBigDig ibase = BC_NUM_BIGDIG_MAX;
1541         BcBigDig env_ibase = BC_NUM_BIGDIG_MAX;
1542         BcBigDig obase = BC_NUM_BIGDIG_MAX;
1543         BcBigDig env_obase = BC_NUM_BIGDIG_MAX;
1544
1545         // We need to know which of stdin, stdout, and stderr are tty's.
1546         ttyin = isatty(STDIN_FILENO);
1547         ttyout = isatty(STDOUT_FILENO);
1548         ttyerr = isatty(STDERR_FILENO);
1549         tty = (ttyin != 0 && ttyout != 0 && ttyerr != 0);
1550
1551         vm->flags |= ttyin ? BC_FLAG_TTYIN : 0;
1552         vm->flags |= tty ? BC_FLAG_TTY : 0;
1553         vm->flags |= ttyin && ttyout ? BC_FLAG_I : 0;
1554
1555         // Set up signals.
1556         bc_vm_sigaction();
1557
1558         // Initialize some vm stuff. This is separate to make things easier for the
1559         // library.
1560         bc_vm_init();
1561
1562         // Explicitly set this in case NULL isn't all zeroes.
1563         vm->file = NULL;
1564
1565         // Set the error messages.
1566         bc_vm_gettext();
1567
1568 #if BC_ENABLE_LINE_LIB
1569
1570         // Initialize the output file buffers.
1571         bc_file_init(&vm->ferr, stderr);
1572         bc_file_init(&vm->fout, stdout);
1573
1574         // Set the input buffer.
1575         vm->buf = output_bufs;
1576
1577 #else // BC_ENABLE_LINE_LIB
1578
1579         // Initialize the output file buffers. They each take portions of the global
1580         // buffer. stdout gets more because it will probably have more data.
1581         bc_file_init(&vm->ferr, STDERR_FILENO, output_bufs + BC_VM_STDOUT_BUF_SIZE,
1582                      BC_VM_STDERR_BUF_SIZE);
1583         bc_file_init(&vm->fout, STDOUT_FILENO, output_bufs, BC_VM_STDOUT_BUF_SIZE);
1584
1585         // Set the input buffer to the rest of the global buffer.
1586         vm->buf = output_bufs + BC_VM_STDOUT_BUF_SIZE + BC_VM_STDERR_BUF_SIZE;
1587 #endif // BC_ENABLE_LINE_LIB
1588
1589         // Set the line length by environment variable.
1590         vm->line_len = (uint16_t) bc_vm_envLen(env_len);
1591
1592         bc_vm_setenvFlag(env_exit, env_exit_def, BC_FLAG_EXPR_EXIT);
1593         bc_vm_setenvFlag(env_clamp, env_clamp_def, BC_FLAG_DIGIT_CLAMP);
1594
1595         // Clear the files and expressions vectors, just in case. This marks them as
1596         // *not* allocated.
1597         bc_vec_clear(&vm->files);
1598         bc_vec_clear(&vm->exprs);
1599
1600 #if !BC_ENABLE_LIBRARY
1601
1602         // Initialize the slab vector.
1603         bc_slabvec_init(&vm->slabs);
1604
1605 #endif // !BC_ENABLE_LIBRARY
1606
1607         // Initialize the program and main parser. These have to be in this order
1608         // because the program has to be initialized first, since a pointer to it is
1609         // passed to the parser.
1610         bc_program_init(&vm->prog);
1611         bc_parse_init(&vm->prs, &vm->prog, BC_PROG_MAIN);
1612
1613         // Set defaults.
1614         vm->flags |= BC_TTY ? BC_FLAG_P | BC_FLAG_R : 0;
1615         vm->flags |= BC_I ? BC_FLAG_Q : 0;
1616
1617 #if BC_ENABLED
1618         if (BC_IS_BC)
1619         {
1620                 // bc checks this environment variable to see if it should run in
1621                 // standard mode.
1622                 char* var = bc_vm_getenv("POSIXLY_CORRECT");
1623
1624                 vm->flags |= BC_FLAG_S * (var != NULL);
1625                 bc_vm_getenvFree(var);
1626
1627                 // Set whether we print the banner or not.
1628                 if (BC_I) bc_vm_setenvFlag("BC_BANNER", BC_DEFAULT_BANNER, BC_FLAG_Q);
1629         }
1630 #endif // BC_ENABLED
1631
1632         // Are we in TTY mode?
1633         if (BC_TTY)
1634         {
1635                 const char* const env_tty = BC_VM_TTY_MODE_STR;
1636                 int env_tty_def = BC_VM_TTY_MODE_DEF;
1637                 const char* const env_prompt = BC_VM_PROMPT_STR;
1638                 int env_prompt_def = BC_VM_PROMPT_DEF;
1639
1640                 // Set flags for TTY mode and prompt.
1641                 bc_vm_setenvFlag(env_tty, env_tty_def, BC_FLAG_TTY);
1642                 bc_vm_setenvFlag(env_prompt, tty ? env_prompt_def : 0, BC_FLAG_P);
1643
1644 #if BC_ENABLE_HISTORY
1645                 // If TTY mode is used, activate history.
1646                 if (BC_TTY) bc_history_init(&vm->history);
1647 #endif // BC_ENABLE_HISTORY
1648         }
1649
1650         // Process environment and command-line arguments.
1651         bc_vm_envArgs(env_args, &env_scale, &env_ibase, &env_obase);
1652         bc_args(argc, argv, true, &scale, &ibase, &obase);
1653
1654         // This section is here because we don't want the math library to stomp on
1655         // the user's given value for scale. And we don't want ibase affecting how
1656         // the scale is interpreted. Also, it's sectioned off just for this comment.
1657         {
1658                 BC_SIG_UNLOCK;
1659
1660                 scale = scale == BC_NUM_BIGDIG_MAX ? env_scale : scale;
1661 #if BC_ENABLED
1662                 // Assign the library value only if it is used and no value was set.
1663                 scale = scale == BC_NUM_BIGDIG_MAX && BC_L ? 20 : scale;
1664 #endif // BC_ENABLED
1665                 obase = obase == BC_NUM_BIGDIG_MAX ? env_obase : obase;
1666                 ibase = ibase == BC_NUM_BIGDIG_MAX ? env_ibase : ibase;
1667
1668                 if (scale != BC_NUM_BIGDIG_MAX)
1669                 {
1670                         bc_program_assignBuiltin(&vm->prog, true, false, scale);
1671                 }
1672
1673                 if (obase != BC_NUM_BIGDIG_MAX)
1674                 {
1675                         bc_program_assignBuiltin(&vm->prog, false, true, obase);
1676                 }
1677
1678                 // This is last to avoid it affecting the value of the others.
1679                 if (ibase != BC_NUM_BIGDIG_MAX)
1680                 {
1681                         bc_program_assignBuiltin(&vm->prog, false, false, ibase);
1682                 }
1683
1684                 BC_SIG_LOCK;
1685         }
1686
1687         // If we are in interactive mode...
1688         if (BC_I)
1689         {
1690                 const char* const env_sigint = BC_VM_SIGINT_RESET_STR;
1691                 int env_sigint_def = BC_VM_SIGINT_RESET_DEF;
1692
1693                 // Set whether we reset on SIGINT or not.
1694                 bc_vm_setenvFlag(env_sigint, env_sigint_def, BC_FLAG_SIGINT);
1695         }
1696
1697 #if BC_ENABLED
1698         // Disable global stacks in POSIX mode.
1699         if (BC_IS_POSIX) vm->flags &= ~(BC_FLAG_G);
1700
1701         // Print the banner if allowed. We have to be in bc, in interactive mode,
1702         // and not be quieted by command-line option or environment variable.
1703         if (BC_IS_BC && BC_I && (vm->flags & BC_FLAG_Q))
1704         {
1705                 bc_vm_info(NULL);
1706                 bc_file_putchar(&vm->fout, bc_flush_none, '\n');
1707                 bc_file_flush(&vm->fout, bc_flush_none);
1708         }
1709 #endif // BC_ENABLED
1710
1711         BC_SIG_UNLOCK;
1712
1713         // Start executing.
1714         bc_vm_exec();
1715 }
1716 #endif // !BC_ENABLE_LIBRARY
1717
1718 void
1719 bc_vm_init(void)
1720 {
1721 #if BC_ENABLE_LIBRARY
1722         BcVm* vm = bcl_getspecific();
1723 #endif // BC_ENABLE_LIBRARY
1724
1725         BC_SIG_ASSERT_LOCKED;
1726
1727 #if !BC_ENABLE_LIBRARY
1728         // Set up the constant zero.
1729         bc_num_setup(&vm->zero, vm->zero_num, BC_VM_ONE_CAP);
1730 #endif // !BC_ENABLE_LIBRARY
1731
1732         // Set up more constant BcNum's.
1733         bc_num_setup(&vm->one, vm->one_num, BC_VM_ONE_CAP);
1734         bc_num_one(&vm->one);
1735
1736         // Set up more constant BcNum's.
1737         // NOLINTNEXTLINE
1738         memcpy(vm->max_num, bc_num_bigdigMax,
1739                bc_num_bigdigMax_size * sizeof(BcDig));
1740         // NOLINTNEXTLINE
1741         memcpy(vm->max2_num, bc_num_bigdigMax2,
1742                bc_num_bigdigMax2_size * sizeof(BcDig));
1743         bc_num_setup(&vm->max, vm->max_num, BC_NUM_BIGDIG_LOG10);
1744         bc_num_setup(&vm->max2, vm->max2_num, BC_NUM_BIGDIG_LOG10);
1745         vm->max.len = bc_num_bigdigMax_size;
1746         vm->max2.len = bc_num_bigdigMax2_size;
1747
1748         // Set up the maxes for the globals.
1749         vm->maxes[BC_PROG_GLOBALS_IBASE] = BC_NUM_MAX_POSIX_IBASE;
1750         vm->maxes[BC_PROG_GLOBALS_OBASE] = BC_MAX_OBASE;
1751         vm->maxes[BC_PROG_GLOBALS_SCALE] = BC_MAX_SCALE;
1752
1753 #if BC_ENABLE_EXTRA_MATH
1754         vm->maxes[BC_PROG_MAX_RAND] = ((BcRand) 0) - 1;
1755 #endif // BC_ENABLE_EXTRA_MATH
1756
1757 #if BC_ENABLED
1758 #if !BC_ENABLE_LIBRARY
1759         // bc has a higher max ibase when it's not in POSIX mode.
1760         if (BC_IS_BC && !BC_IS_POSIX)
1761 #endif // !BC_ENABLE_LIBRARY
1762         {
1763                 vm->maxes[BC_PROG_GLOBALS_IBASE] = BC_NUM_MAX_IBASE;
1764         }
1765 #endif // BC_ENABLED
1766 }
1767
1768 #if BC_ENABLE_LIBRARY
1769 void
1770 bc_vm_atexit(void)
1771 {
1772 #if BC_DEBUG
1773 #if BC_ENABLE_LIBRARY
1774         BcVm* vm = bcl_getspecific();
1775 #endif // BC_ENABLE_LIBRARY
1776 #endif // BC_DEBUG
1777
1778         bc_vm_shutdown();
1779
1780 #if BC_DEBUG
1781         bc_vec_free(&vm->jmp_bufs);
1782 #endif // BC_DEBUG
1783 }
1784 #else // BC_ENABLE_LIBRARY
1785 int
1786 bc_vm_atexit(int status)
1787 {
1788         // Set the status correctly.
1789         int s = BC_STATUS_IS_ERROR(status) ? status : BC_STATUS_SUCCESS;
1790
1791         bc_vm_shutdown();
1792
1793 #if BC_DEBUG
1794         bc_vec_free(&vm->jmp_bufs);
1795 #endif // BC_DEBUG
1796
1797         return s;
1798 }
1799 #endif // BC_ENABLE_LIBRARY