xref: /freebsd/contrib/bc/src/vm.c (revision 50696a6e8cbfdbf4a0d00f2f85f1951aa0d8e23d)
1252884aeSStefan Eßer /*
2252884aeSStefan Eßer  * *****************************************************************************
3252884aeSStefan Eßer  *
43aa99676SStefan Eßer  * SPDX-License-Identifier: BSD-2-Clause
5252884aeSStefan Eßer  *
63aa99676SStefan Eßer  * Copyright (c) 2018-2020 Gavin D. Howard and contributors.
7252884aeSStefan Eßer  *
8252884aeSStefan Eßer  * Redistribution and use in source and binary forms, with or without
9252884aeSStefan Eßer  * modification, are permitted provided that the following conditions are met:
10252884aeSStefan Eßer  *
11252884aeSStefan Eßer  * * Redistributions of source code must retain the above copyright notice, this
12252884aeSStefan Eßer  *   list of conditions and the following disclaimer.
13252884aeSStefan Eßer  *
14252884aeSStefan Eßer  * * Redistributions in binary form must reproduce the above copyright notice,
15252884aeSStefan Eßer  *   this list of conditions and the following disclaimer in the documentation
16252884aeSStefan Eßer  *   and/or other materials provided with the distribution.
17252884aeSStefan Eßer  *
18252884aeSStefan Eßer  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19252884aeSStefan Eßer  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20252884aeSStefan Eßer  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21252884aeSStefan Eßer  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
22252884aeSStefan Eßer  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23252884aeSStefan Eßer  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24252884aeSStefan Eßer  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25252884aeSStefan Eßer  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26252884aeSStefan Eßer  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27252884aeSStefan Eßer  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28252884aeSStefan Eßer  * POSSIBILITY OF SUCH DAMAGE.
29252884aeSStefan Eßer  *
30252884aeSStefan Eßer  * *****************************************************************************
31252884aeSStefan Eßer  *
32252884aeSStefan Eßer  * Code common to all of bc and dc.
33252884aeSStefan Eßer  *
34252884aeSStefan Eßer  */
35252884aeSStefan Eßer 
36252884aeSStefan Eßer #include <assert.h>
37252884aeSStefan Eßer #include <ctype.h>
38252884aeSStefan Eßer #include <errno.h>
39252884aeSStefan Eßer #include <stdarg.h>
40252884aeSStefan Eßer #include <string.h>
41252884aeSStefan Eßer 
42252884aeSStefan Eßer #include <signal.h>
43252884aeSStefan Eßer 
44252884aeSStefan Eßer #include <setjmp.h>
45252884aeSStefan Eßer 
46252884aeSStefan Eßer #ifndef _WIN32
47252884aeSStefan Eßer 
48252884aeSStefan Eßer #include <sys/types.h>
49252884aeSStefan Eßer #include <unistd.h>
50252884aeSStefan Eßer 
51252884aeSStefan Eßer #else // _WIN32
52252884aeSStefan Eßer 
53252884aeSStefan Eßer #define WIN32_LEAN_AND_MEAN
54252884aeSStefan Eßer #include <windows.h>
55252884aeSStefan Eßer #include <io.h>
56252884aeSStefan Eßer 
57252884aeSStefan Eßer #endif // _WIN32
58252884aeSStefan Eßer 
59252884aeSStefan Eßer #include <vector.h>
60252884aeSStefan Eßer #include <args.h>
61252884aeSStefan Eßer #include <vm.h>
62252884aeSStefan Eßer #include <read.h>
63252884aeSStefan Eßer #include <bc.h>
64252884aeSStefan Eßer 
65*50696a6eSStefan Eßer char output_bufs[BC_VM_BUF_SIZE];
66*50696a6eSStefan Eßer BcVm vm;
67*50696a6eSStefan Eßer 
68252884aeSStefan Eßer #if BC_DEBUG_CODE
69252884aeSStefan Eßer BC_NORETURN void bc_vm_jmp(const char* f) {
70252884aeSStefan Eßer #else // BC_DEBUG_CODE
71252884aeSStefan Eßer BC_NORETURN void bc_vm_jmp(void) {
72252884aeSStefan Eßer #endif
73252884aeSStefan Eßer 
743aa99676SStefan Eßer 	assert(BC_SIG_EXC);
75252884aeSStefan Eßer 
76252884aeSStefan Eßer 	BC_SIG_MAYLOCK;
77252884aeSStefan Eßer 
78252884aeSStefan Eßer #if BC_DEBUG_CODE
79252884aeSStefan Eßer 	bc_file_puts(&vm.ferr, "Longjmp: ");
80252884aeSStefan Eßer 	bc_file_puts(&vm.ferr, f);
81252884aeSStefan Eßer 	bc_file_putchar(&vm.ferr, '\n');
82252884aeSStefan Eßer 	bc_file_flush(&vm.ferr);
83252884aeSStefan Eßer #endif // BC_DEBUG_CODE
84252884aeSStefan Eßer 
85252884aeSStefan Eßer #ifndef NDEBUG
86252884aeSStefan Eßer 	assert(vm.jmp_bufs.len - (size_t) vm.sig_pop);
87252884aeSStefan Eßer #endif // NDEBUG
88252884aeSStefan Eßer 
89*50696a6eSStefan Eßer 	if (vm.jmp_bufs.len == 0) abort();
90252884aeSStefan Eßer 	if (vm.sig_pop) bc_vec_pop(&vm.jmp_bufs);
91252884aeSStefan Eßer 	else vm.sig_pop = 1;
92252884aeSStefan Eßer 
93252884aeSStefan Eßer 	siglongjmp(*((sigjmp_buf*) bc_vec_top(&vm.jmp_bufs)), 1);
94252884aeSStefan Eßer }
95252884aeSStefan Eßer 
96*50696a6eSStefan Eßer #if !BC_ENABLE_LIBRARY
97252884aeSStefan Eßer static void bc_vm_sig(int sig) {
98252884aeSStefan Eßer 
99252884aeSStefan Eßer 	// There is already a signal in flight.
100252884aeSStefan Eßer 	if (vm.status == (sig_atomic_t) BC_STATUS_QUIT || vm.sig) {
101252884aeSStefan Eßer 		if (!BC_TTY || sig != SIGINT) vm.status = BC_STATUS_QUIT;
102252884aeSStefan Eßer 		return;
103252884aeSStefan Eßer 	}
104252884aeSStefan Eßer 
105252884aeSStefan Eßer 	if (BC_TTY && sig == SIGINT) {
106252884aeSStefan Eßer 
107252884aeSStefan Eßer 		int err = errno;
108252884aeSStefan Eßer 
109252884aeSStefan Eßer 		if (write(STDOUT_FILENO, vm.sigmsg, vm.siglen) != (ssize_t) vm.siglen)
110252884aeSStefan Eßer 			vm.status = BC_STATUS_ERROR_FATAL;
111252884aeSStefan Eßer 		else vm.sig = 1;
112252884aeSStefan Eßer 
113252884aeSStefan Eßer 		errno = err;
114252884aeSStefan Eßer 	}
115252884aeSStefan Eßer 	else vm.status = BC_STATUS_QUIT;
116252884aeSStefan Eßer 
117252884aeSStefan Eßer 	assert(vm.jmp_bufs.len);
118252884aeSStefan Eßer 
119252884aeSStefan Eßer 	if (!vm.sig_lock) BC_VM_JMP;
120252884aeSStefan Eßer }
121252884aeSStefan Eßer 
122252884aeSStefan Eßer void bc_vm_info(const char* const help) {
123252884aeSStefan Eßer 
124252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
125252884aeSStefan Eßer 
126252884aeSStefan Eßer 	bc_file_puts(&vm.fout, vm.name);
127252884aeSStefan Eßer 	bc_file_putchar(&vm.fout, ' ');
128252884aeSStefan Eßer 	bc_file_puts(&vm.fout, BC_VERSION);
129252884aeSStefan Eßer 	bc_file_putchar(&vm.fout, '\n');
130252884aeSStefan Eßer 	bc_file_puts(&vm.fout, bc_copyright);
131252884aeSStefan Eßer 
132252884aeSStefan Eßer 	if (help) {
133252884aeSStefan Eßer 		bc_file_putchar(&vm.fout, '\n');
134252884aeSStefan Eßer 		bc_file_printf(&vm.fout, help, vm.name, vm.name);
135252884aeSStefan Eßer 	}
136252884aeSStefan Eßer 
137252884aeSStefan Eßer 	bc_file_flush(&vm.fout);
138252884aeSStefan Eßer }
139*50696a6eSStefan Eßer #endif // !BC_ENABLE_LIBRARY
140252884aeSStefan Eßer 
141*50696a6eSStefan Eßer #if BC_ENABLE_LIBRARY
142*50696a6eSStefan Eßer void bc_vm_handleError(BcErr e) {
143*50696a6eSStefan Eßer 
144*50696a6eSStefan Eßer 	assert(e < BC_ERR_NELEMS);
145*50696a6eSStefan Eßer 	assert(!vm.sig_pop);
146*50696a6eSStefan Eßer 
147*50696a6eSStefan Eßer 	BC_SIG_LOCK;
148*50696a6eSStefan Eßer 
149*50696a6eSStefan Eßer 	if (e <= BC_ERR_MATH_DIVIDE_BY_ZERO) {
150*50696a6eSStefan Eßer 		vm.err = (BclError) (e - BC_ERR_MATH_NEGATIVE +
151*50696a6eSStefan Eßer 		                     BCL_ERROR_MATH_NEGATIVE);
152*50696a6eSStefan Eßer 	}
153*50696a6eSStefan Eßer 	else if (vm.abrt) abort();
154*50696a6eSStefan Eßer 	else if (e == BC_ERR_FATAL_ALLOC_ERR) vm.err = BCL_ERROR_FATAL_ALLOC_ERR;
155*50696a6eSStefan Eßer 	else vm.err = BCL_ERROR_FATAL_UNKNOWN_ERR;
156*50696a6eSStefan Eßer 
157*50696a6eSStefan Eßer 	BC_VM_JMP;
158*50696a6eSStefan Eßer }
159*50696a6eSStefan Eßer #else // BC_ENABLE_LIBRARY
160*50696a6eSStefan Eßer void bc_vm_handleError(BcErr e, size_t line, ...) {
161252884aeSStefan Eßer 
162252884aeSStefan Eßer 	BcStatus s;
163252884aeSStefan Eßer 	va_list args;
164252884aeSStefan Eßer 	uchar id = bc_err_ids[e];
165252884aeSStefan Eßer 	const char* err_type = vm.err_ids[id];
166252884aeSStefan Eßer 	sig_atomic_t lock;
167252884aeSStefan Eßer 
168*50696a6eSStefan Eßer 	assert(e < BC_ERR_NELEMS);
169252884aeSStefan Eßer 	assert(!vm.sig_pop);
170252884aeSStefan Eßer 
171252884aeSStefan Eßer #if BC_ENABLED
172*50696a6eSStefan Eßer 	if (!BC_S && e >= BC_ERR_POSIX_START) {
173252884aeSStefan Eßer 		if (BC_W) {
174252884aeSStefan Eßer 			// Make sure to not return an error.
175252884aeSStefan Eßer 			id = UCHAR_MAX;
176252884aeSStefan Eßer 			err_type = vm.err_ids[BC_ERR_IDX_WARN];
177252884aeSStefan Eßer 		}
178252884aeSStefan Eßer 		else return;
179252884aeSStefan Eßer 	}
180252884aeSStefan Eßer #endif // BC_ENABLED
181252884aeSStefan Eßer 
182252884aeSStefan Eßer 	BC_SIG_TRYLOCK(lock);
183252884aeSStefan Eßer 
184252884aeSStefan Eßer 	// Make sure all of stdout is written first.
185252884aeSStefan Eßer 	s = bc_file_flushErr(&vm.fout);
186252884aeSStefan Eßer 
187252884aeSStefan Eßer 	if (BC_ERR(s == BC_STATUS_ERROR_FATAL)) {
188252884aeSStefan Eßer 		vm.status = (sig_atomic_t) s;
189252884aeSStefan Eßer 		BC_VM_JMP;
190252884aeSStefan Eßer 	}
191252884aeSStefan Eßer 
192252884aeSStefan Eßer 	va_start(args, line);
193252884aeSStefan Eßer 	bc_file_putchar(&vm.ferr, '\n');
194252884aeSStefan Eßer 	bc_file_puts(&vm.ferr, err_type);
195252884aeSStefan Eßer 	bc_file_putchar(&vm.ferr, ' ');
196252884aeSStefan Eßer 	bc_file_vprintf(&vm.ferr, vm.err_msgs[e], args);
197252884aeSStefan Eßer 	va_end(args);
198252884aeSStefan Eßer 
199252884aeSStefan Eßer 	if (BC_NO_ERR(vm.file)) {
200252884aeSStefan Eßer 
201252884aeSStefan Eßer 		// This is the condition for parsing vs runtime.
202252884aeSStefan Eßer 		// If line is not 0, it is parsing.
203252884aeSStefan Eßer 		if (line) {
204252884aeSStefan Eßer 			bc_file_puts(&vm.ferr, "\n    ");
205252884aeSStefan Eßer 			bc_file_puts(&vm.ferr, vm.file);
206252884aeSStefan Eßer 			bc_file_printf(&vm.ferr, bc_err_line, line);
207252884aeSStefan Eßer 		}
208252884aeSStefan Eßer 		else {
209252884aeSStefan Eßer 
210252884aeSStefan Eßer 			BcInstPtr *ip = bc_vec_item_rev(&vm.prog.stack, 0);
211252884aeSStefan Eßer 			BcFunc *f = bc_vec_item(&vm.prog.fns, ip->func);
212252884aeSStefan Eßer 
213252884aeSStefan Eßer 			bc_file_puts(&vm.ferr, "\n    ");
214252884aeSStefan Eßer 			bc_file_puts(&vm.ferr, vm.func_header);
215252884aeSStefan Eßer 			bc_file_putchar(&vm.ferr, ' ');
216252884aeSStefan Eßer 			bc_file_puts(&vm.ferr, f->name);
217252884aeSStefan Eßer 
2183aa99676SStefan Eßer #if BC_ENABLED
219252884aeSStefan Eßer 			if (BC_IS_BC && ip->func != BC_PROG_MAIN &&
220252884aeSStefan Eßer 			    ip->func != BC_PROG_READ)
221252884aeSStefan Eßer 			{
222252884aeSStefan Eßer 				bc_file_puts(&vm.ferr, "()");
223252884aeSStefan Eßer 			}
2243aa99676SStefan Eßer #endif // BC_ENABLED
225252884aeSStefan Eßer 		}
226252884aeSStefan Eßer 	}
227252884aeSStefan Eßer 
228252884aeSStefan Eßer 	bc_file_puts(&vm.ferr, "\n\n");
229252884aeSStefan Eßer 
230252884aeSStefan Eßer 	s = bc_file_flushErr(&vm.ferr);
231252884aeSStefan Eßer 
232252884aeSStefan Eßer 	vm.status = s == BC_STATUS_ERROR_FATAL ?
233252884aeSStefan Eßer 	    (sig_atomic_t) s : (sig_atomic_t) (uchar) (id + 1);
234252884aeSStefan Eßer 
235252884aeSStefan Eßer 	if (BC_ERR(vm.status)) BC_VM_JMP;
236252884aeSStefan Eßer 
237252884aeSStefan Eßer 	BC_SIG_TRYUNLOCK(lock);
238252884aeSStefan Eßer }
239252884aeSStefan Eßer 
240252884aeSStefan Eßer static void bc_vm_envArgs(const char* const env_args_name) {
241252884aeSStefan Eßer 
242252884aeSStefan Eßer 	char *env_args = getenv(env_args_name), *buf, *start;
243252884aeSStefan Eßer 	char instr = '\0';
244252884aeSStefan Eßer 
245252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
246252884aeSStefan Eßer 
247252884aeSStefan Eßer 	if (env_args == NULL) return;
248252884aeSStefan Eßer 
249252884aeSStefan Eßer 	start = buf = vm.env_args_buffer = bc_vm_strdup(env_args);
250252884aeSStefan Eßer 
251252884aeSStefan Eßer 	assert(buf != NULL);
252252884aeSStefan Eßer 
253252884aeSStefan Eßer 	bc_vec_init(&vm.env_args, sizeof(char*), NULL);
254252884aeSStefan Eßer 	bc_vec_push(&vm.env_args, &env_args_name);
255252884aeSStefan Eßer 
256252884aeSStefan Eßer 	while (*buf) {
257252884aeSStefan Eßer 
258252884aeSStefan Eßer 		if (!isspace(*buf)) {
259252884aeSStefan Eßer 
260252884aeSStefan Eßer 			if (*buf == '"' || *buf == '\'') {
261252884aeSStefan Eßer 
262252884aeSStefan Eßer 				instr = *buf;
263252884aeSStefan Eßer 				buf += 1;
264252884aeSStefan Eßer 
265252884aeSStefan Eßer 				if (*buf == instr) {
266252884aeSStefan Eßer 					instr = '\0';
267252884aeSStefan Eßer 					buf += 1;
268252884aeSStefan Eßer 					continue;
269252884aeSStefan Eßer 				}
270252884aeSStefan Eßer 			}
271252884aeSStefan Eßer 
272252884aeSStefan Eßer 			bc_vec_push(&vm.env_args, &buf);
273252884aeSStefan Eßer 
274252884aeSStefan Eßer 			while (*buf && ((!instr && !isspace(*buf)) ||
275252884aeSStefan Eßer 			                (instr && *buf != instr)))
276252884aeSStefan Eßer 			{
277252884aeSStefan Eßer 				buf += 1;
278252884aeSStefan Eßer 			}
279252884aeSStefan Eßer 
280252884aeSStefan Eßer 			if (*buf) {
281252884aeSStefan Eßer 
282252884aeSStefan Eßer 				if (instr) instr = '\0';
283252884aeSStefan Eßer 
284252884aeSStefan Eßer 				*buf = '\0';
285252884aeSStefan Eßer 				buf += 1;
286252884aeSStefan Eßer 				start = buf;
287252884aeSStefan Eßer 			}
288*50696a6eSStefan Eßer 			else if (instr) bc_vm_error(BC_ERR_FATAL_OPTION, 0, start);
289252884aeSStefan Eßer 		}
290252884aeSStefan Eßer 		else buf += 1;
291252884aeSStefan Eßer 	}
292252884aeSStefan Eßer 
293252884aeSStefan Eßer 	// Make sure to push a NULL pointer at the end.
294252884aeSStefan Eßer 	buf = NULL;
295252884aeSStefan Eßer 	bc_vec_push(&vm.env_args, &buf);
296252884aeSStefan Eßer 
297252884aeSStefan Eßer 	bc_args((int) vm.env_args.len - 1, bc_vec_item(&vm.env_args, 0));
298252884aeSStefan Eßer }
299252884aeSStefan Eßer 
300252884aeSStefan Eßer static size_t bc_vm_envLen(const char *var) {
301252884aeSStefan Eßer 
302252884aeSStefan Eßer 	char *lenv = getenv(var);
303252884aeSStefan Eßer 	size_t i, len = BC_NUM_PRINT_WIDTH;
304252884aeSStefan Eßer 	int num;
305252884aeSStefan Eßer 
306252884aeSStefan Eßer 	if (lenv == NULL) return len;
307252884aeSStefan Eßer 
308252884aeSStefan Eßer 	len = strlen(lenv);
309252884aeSStefan Eßer 
310252884aeSStefan Eßer 	for (num = 1, i = 0; num && i < len; ++i) num = isdigit(lenv[i]);
311252884aeSStefan Eßer 
312252884aeSStefan Eßer 	if (num) {
313252884aeSStefan Eßer 		len = (size_t) atoi(lenv) - 1;
314252884aeSStefan Eßer 		if (len < 2 || len >= UINT16_MAX) len = BC_NUM_PRINT_WIDTH;
315252884aeSStefan Eßer 	}
316252884aeSStefan Eßer 	else len = BC_NUM_PRINT_WIDTH;
317252884aeSStefan Eßer 
318252884aeSStefan Eßer 	return len;
319252884aeSStefan Eßer }
320*50696a6eSStefan Eßer #endif // BC_ENABLE_LIBRARY
321252884aeSStefan Eßer 
322252884aeSStefan Eßer void bc_vm_shutdown(void) {
323252884aeSStefan Eßer 
324252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
325252884aeSStefan Eßer 
326252884aeSStefan Eßer #if BC_ENABLE_NLS
327252884aeSStefan Eßer 	if (vm.catalog != BC_VM_INVALID_CATALOG) catclose(vm.catalog);
328252884aeSStefan Eßer #endif // BC_ENABLE_NLS
329252884aeSStefan Eßer 
330252884aeSStefan Eßer #if BC_ENABLE_HISTORY
331252884aeSStefan Eßer 	// This must always run to ensure that the terminal is back to normal.
332252884aeSStefan Eßer 	if (BC_TTY) bc_history_free(&vm.history);
333252884aeSStefan Eßer #endif // BC_ENABLE_HISTORY
334252884aeSStefan Eßer 
335252884aeSStefan Eßer #ifndef NDEBUG
336*50696a6eSStefan Eßer #if !BC_ENABLE_LIBRARY
337252884aeSStefan Eßer 	bc_vec_free(&vm.env_args);
338252884aeSStefan Eßer 	free(vm.env_args_buffer);
339252884aeSStefan Eßer 	bc_vec_free(&vm.files);
340252884aeSStefan Eßer 	bc_vec_free(&vm.exprs);
341252884aeSStefan Eßer 
342252884aeSStefan Eßer 	bc_program_free(&vm.prog);
343252884aeSStefan Eßer 	bc_parse_free(&vm.prs);
344*50696a6eSStefan Eßer #endif // !BC_ENABLE_LIBRARY
345252884aeSStefan Eßer 
346*50696a6eSStefan Eßer 	bc_vm_freeTemps();
347252884aeSStefan Eßer 	bc_vec_free(&vm.temps);
348252884aeSStefan Eßer #endif // NDEBUG
349252884aeSStefan Eßer 
350*50696a6eSStefan Eßer #if !BC_ENABLE_LIBRARY
351252884aeSStefan Eßer 	bc_file_free(&vm.fout);
352252884aeSStefan Eßer 	bc_file_free(&vm.ferr);
353*50696a6eSStefan Eßer #endif // !BC_ENABLE_LIBRARY
354252884aeSStefan Eßer }
355252884aeSStefan Eßer 
356*50696a6eSStefan Eßer #if !defined(NDEBUG) || BC_ENABLE_LIBRARY
357*50696a6eSStefan Eßer void bc_vm_freeTemps(void) {
358*50696a6eSStefan Eßer 
359*50696a6eSStefan Eßer 	size_t i;
360*50696a6eSStefan Eßer 
361*50696a6eSStefan Eßer 	for (i = 0; i < vm.temps.len; ++i) {
362*50696a6eSStefan Eßer 		free(((BcNum*) bc_vec_item(&vm.temps, i))->num);
363*50696a6eSStefan Eßer 	}
364*50696a6eSStefan Eßer }
365*50696a6eSStefan Eßer #endif // !defined(NDEBUG) || BC_ENABLE_LIBRARY
366*50696a6eSStefan Eßer 
3673aa99676SStefan Eßer inline size_t bc_vm_arraySize(size_t n, size_t size) {
368252884aeSStefan Eßer 	size_t res = n * size;
369252884aeSStefan Eßer 	if (BC_ERR(res >= SIZE_MAX || (n != 0 && res / n != size)))
370*50696a6eSStefan Eßer 		bc_vm_err(BC_ERR_FATAL_ALLOC_ERR);
371252884aeSStefan Eßer 	return res;
372252884aeSStefan Eßer }
373252884aeSStefan Eßer 
3743aa99676SStefan Eßer inline size_t bc_vm_growSize(size_t a, size_t b) {
375252884aeSStefan Eßer 	size_t res = a + b;
376252884aeSStefan Eßer 	if (BC_ERR(res >= SIZE_MAX || res < a || res < b))
377*50696a6eSStefan Eßer 		bc_vm_err(BC_ERR_FATAL_ALLOC_ERR);
378252884aeSStefan Eßer 	return res;
379252884aeSStefan Eßer }
380252884aeSStefan Eßer 
381252884aeSStefan Eßer void* bc_vm_malloc(size_t n) {
382252884aeSStefan Eßer 
383252884aeSStefan Eßer 	void* ptr;
384252884aeSStefan Eßer 
385252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
386252884aeSStefan Eßer 
387252884aeSStefan Eßer 	ptr = malloc(n);
388252884aeSStefan Eßer 
389*50696a6eSStefan Eßer 	if (BC_ERR(ptr == NULL)) bc_vm_err(BC_ERR_FATAL_ALLOC_ERR);
390252884aeSStefan Eßer 
391252884aeSStefan Eßer 	return ptr;
392252884aeSStefan Eßer }
393252884aeSStefan Eßer 
394252884aeSStefan Eßer void* bc_vm_realloc(void *ptr, size_t n) {
395252884aeSStefan Eßer 
396252884aeSStefan Eßer 	void* temp;
397252884aeSStefan Eßer 
398252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
399252884aeSStefan Eßer 
400252884aeSStefan Eßer 	temp = realloc(ptr, n);
401252884aeSStefan Eßer 
402*50696a6eSStefan Eßer 	if (BC_ERR(temp == NULL)) bc_vm_err(BC_ERR_FATAL_ALLOC_ERR);
403252884aeSStefan Eßer 
404252884aeSStefan Eßer 	return temp;
405252884aeSStefan Eßer }
406252884aeSStefan Eßer 
407252884aeSStefan Eßer char* bc_vm_strdup(const char *str) {
408252884aeSStefan Eßer 
409252884aeSStefan Eßer 	char *s;
410252884aeSStefan Eßer 
411252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
412252884aeSStefan Eßer 
413252884aeSStefan Eßer 	s = strdup(str);
414252884aeSStefan Eßer 
415*50696a6eSStefan Eßer 	if (BC_ERR(!s)) bc_vm_err(BC_ERR_FATAL_ALLOC_ERR);
416252884aeSStefan Eßer 
417252884aeSStefan Eßer 	return s;
418252884aeSStefan Eßer }
419252884aeSStefan Eßer 
420*50696a6eSStefan Eßer #if !BC_ENABLE_LIBRARY
421252884aeSStefan Eßer void bc_vm_printf(const char *fmt, ...) {
422252884aeSStefan Eßer 
423252884aeSStefan Eßer 	va_list args;
424252884aeSStefan Eßer 
425252884aeSStefan Eßer 	BC_SIG_LOCK;
426252884aeSStefan Eßer 
427252884aeSStefan Eßer 	va_start(args, fmt);
428252884aeSStefan Eßer 	bc_file_vprintf(&vm.fout, fmt, args);
429252884aeSStefan Eßer 	va_end(args);
430252884aeSStefan Eßer 
431252884aeSStefan Eßer 	vm.nchars = 0;
432252884aeSStefan Eßer 
433252884aeSStefan Eßer 	BC_SIG_UNLOCK;
434252884aeSStefan Eßer }
435*50696a6eSStefan Eßer #endif // !BC_ENABLE_LIBRARY
436252884aeSStefan Eßer 
437252884aeSStefan Eßer void bc_vm_putchar(int c) {
438*50696a6eSStefan Eßer #if BC_ENABLE_LIBRARY
439*50696a6eSStefan Eßer 	bc_vec_pushByte(&vm.out, (uchar) c);
440*50696a6eSStefan Eßer #else // BC_ENABLE_LIBRARY
441252884aeSStefan Eßer 	bc_file_putchar(&vm.fout, (uchar) c);
442252884aeSStefan Eßer 	vm.nchars = (c == '\n' ? 0 : vm.nchars + 1);
443*50696a6eSStefan Eßer #endif // BC_ENABLE_LIBRARY
444252884aeSStefan Eßer }
445252884aeSStefan Eßer 
446*50696a6eSStefan Eßer #if !BC_ENABLE_LIBRARY
447252884aeSStefan Eßer static void bc_vm_clean(void) {
448252884aeSStefan Eßer 
4493aa99676SStefan Eßer 	BcVec *fns = &vm.prog.fns;
450252884aeSStefan Eßer 	BcFunc *f = bc_vec_item(fns, BC_PROG_MAIN);
4513aa99676SStefan Eßer 	BcInstPtr *ip = bc_vec_item(&vm.prog.stack, 0);
4523aa99676SStefan Eßer 	bool good = ((vm.status && vm.status != BC_STATUS_QUIT) || vm.sig);
453252884aeSStefan Eßer 
454252884aeSStefan Eßer 	if (good) bc_program_reset(&vm.prog);
455252884aeSStefan Eßer 
456252884aeSStefan Eßer #if BC_ENABLED
457252884aeSStefan Eßer 	if (good && BC_IS_BC) good = !BC_PARSE_NO_EXEC(&vm.prs);
458252884aeSStefan Eßer #endif // BC_ENABLED
459252884aeSStefan Eßer 
460252884aeSStefan Eßer #if DC_ENABLED
4613aa99676SStefan Eßer 	if (BC_IS_DC) {
462252884aeSStefan Eßer 
463252884aeSStefan Eßer 		size_t i;
464252884aeSStefan Eßer 
4653aa99676SStefan Eßer 		good = true;
466252884aeSStefan Eßer 
4673aa99676SStefan Eßer 		for (i = 0; good && i < vm.prog.results.len; ++i) {
4683aa99676SStefan Eßer 			BcResult *r = (BcResult*) bc_vec_item(&vm.prog.results, i);
4693aa99676SStefan Eßer 			good = BC_VM_SAFE_RESULT(r);
470252884aeSStefan Eßer 		}
471252884aeSStefan Eßer 	}
472252884aeSStefan Eßer #endif // DC_ENABLED
473252884aeSStefan Eßer 
474252884aeSStefan Eßer 	// If this condition is true, we can get rid of strings,
475252884aeSStefan Eßer 	// constants, and code. This is an idea from busybox.
4763aa99676SStefan Eßer 	if (good && vm.prog.stack.len == 1 && ip->idx == f->code.len) {
4773aa99676SStefan Eßer 
478252884aeSStefan Eßer #if BC_ENABLED
479252884aeSStefan Eßer 		if (BC_IS_BC) {
480252884aeSStefan Eßer 			bc_vec_npop(&f->labels, f->labels.len);
481252884aeSStefan Eßer 			bc_vec_npop(&f->strs, f->strs.len);
4823aa99676SStefan Eßer 			bc_vec_npop(&f->consts, f->consts.len);
483252884aeSStefan Eßer 		}
484252884aeSStefan Eßer #endif // BC_ENABLED
4853aa99676SStefan Eßer 
4863aa99676SStefan Eßer #if DC_ENABLED
4873aa99676SStefan Eßer 		// Note to self: you cannot delete strings and functions. Deal with it.
4883aa99676SStefan Eßer 		if (BC_IS_DC) bc_vec_npop(vm.prog.consts, vm.prog.consts->len);
4893aa99676SStefan Eßer #endif // DC_ENABLED
4903aa99676SStefan Eßer 
491252884aeSStefan Eßer 		bc_vec_npop(&f->code, f->code.len);
4923aa99676SStefan Eßer 
493252884aeSStefan Eßer 		ip->idx = 0;
494252884aeSStefan Eßer 	}
495252884aeSStefan Eßer }
496252884aeSStefan Eßer 
4975d934bc0SStefan Eßer static void bc_vm_process(const char *text) {
498252884aeSStefan Eßer 
499252884aeSStefan Eßer 	bc_parse_text(&vm.prs, text);
500252884aeSStefan Eßer 
501252884aeSStefan Eßer 	do {
502252884aeSStefan Eßer 
503252884aeSStefan Eßer #if BC_ENABLED
504252884aeSStefan Eßer 		if (vm.prs.l.t == BC_LEX_KW_DEFINE) vm.parse(&vm.prs);
505252884aeSStefan Eßer #endif // BC_ENABLED
506252884aeSStefan Eßer 
507252884aeSStefan Eßer 		while (BC_PARSE_CAN_PARSE(vm.prs)) vm.parse(&vm.prs);
508252884aeSStefan Eßer 
509119656bcSStefan Eßer 		if(BC_IS_DC || !BC_PARSE_NO_EXEC(&vm.prs)) bc_program_exec(&vm.prog);
5103aa99676SStefan Eßer 
5113aa99676SStefan Eßer 		assert(BC_IS_DC || vm.prog.results.len == 0);
5123aa99676SStefan Eßer 
513252884aeSStefan Eßer 		if (BC_I) bc_file_flush(&vm.fout);
514252884aeSStefan Eßer 
515252884aeSStefan Eßer 	} while (vm.prs.l.t != BC_LEX_EOF);
516252884aeSStefan Eßer }
517252884aeSStefan Eßer 
5185d934bc0SStefan Eßer #if BC_ENABLED
5195d934bc0SStefan Eßer static void bc_vm_endif(void) {
5205d934bc0SStefan Eßer 
5215d934bc0SStefan Eßer 	size_t i;
5225d934bc0SStefan Eßer 	bool good;
5235d934bc0SStefan Eßer 
5245d934bc0SStefan Eßer 	if (BC_NO_ERR(!BC_PARSE_NO_EXEC(&vm.prs))) return;
5255d934bc0SStefan Eßer 
5265d934bc0SStefan Eßer 	good = true;
5275d934bc0SStefan Eßer 
5285d934bc0SStefan Eßer 	for (i = 0; good && i < vm.prs.flags.len; ++i) {
5295d934bc0SStefan Eßer 		uint16_t flag = *((uint16_t*) bc_vec_item(&vm.prs.flags, i));
5305d934bc0SStefan Eßer 		good = ((flag & BC_PARSE_FLAG_BRACE) != BC_PARSE_FLAG_BRACE);
5315d934bc0SStefan Eßer 	}
5325d934bc0SStefan Eßer 
5335d934bc0SStefan Eßer 	if (good) {
5345d934bc0SStefan Eßer 		while (BC_PARSE_IF_END(&vm.prs)) bc_vm_process("else {}");
5355d934bc0SStefan Eßer 	}
536*50696a6eSStefan Eßer 	else bc_parse_err(&vm.prs, BC_ERR_PARSE_BLOCK);
5375d934bc0SStefan Eßer }
5385d934bc0SStefan Eßer #endif // BC_ENABLED
5395d934bc0SStefan Eßer 
540252884aeSStefan Eßer static void bc_vm_file(const char *file) {
541252884aeSStefan Eßer 
542252884aeSStefan Eßer 	char *data = NULL;
543252884aeSStefan Eßer 
544252884aeSStefan Eßer 	assert(!vm.sig_pop);
545252884aeSStefan Eßer 
546252884aeSStefan Eßer 	bc_lex_file(&vm.prs.l, file);
547252884aeSStefan Eßer 
548252884aeSStefan Eßer 	BC_SIG_LOCK;
549252884aeSStefan Eßer 
550252884aeSStefan Eßer 	bc_read_file(file, &data);
551252884aeSStefan Eßer 
552252884aeSStefan Eßer 	BC_SETJMP_LOCKED(err);
553252884aeSStefan Eßer 
554252884aeSStefan Eßer 	BC_SIG_UNLOCK;
555252884aeSStefan Eßer 
5565d934bc0SStefan Eßer 	bc_vm_process(data);
557252884aeSStefan Eßer 
558252884aeSStefan Eßer #if BC_ENABLED
5595d934bc0SStefan Eßer 	if (BC_IS_BC) bc_vm_endif();
560252884aeSStefan Eßer #endif // BC_ENABLED
561252884aeSStefan Eßer 
562252884aeSStefan Eßer err:
563252884aeSStefan Eßer 	BC_SIG_MAYLOCK;
564252884aeSStefan Eßer 
565252884aeSStefan Eßer 	free(data);
566252884aeSStefan Eßer 	bc_vm_clean();
567252884aeSStefan Eßer 
568252884aeSStefan Eßer 	// bc_program_reset(), called by bc_vm_clean(), resets the status.
569252884aeSStefan Eßer 	// We want it to clear the sig_pop variable in case it was set.
570252884aeSStefan Eßer 	if (vm.status == (sig_atomic_t) BC_STATUS_SUCCESS) BC_LONGJMP_STOP;
571252884aeSStefan Eßer 
572252884aeSStefan Eßer 	BC_LONGJMP_CONT;
573252884aeSStefan Eßer }
574252884aeSStefan Eßer 
575252884aeSStefan Eßer static void bc_vm_stdin(void) {
576252884aeSStefan Eßer 
577252884aeSStefan Eßer 	BcStatus s;
578252884aeSStefan Eßer 	BcVec buf, buffer;
579252884aeSStefan Eßer 	size_t string = 0;
580252884aeSStefan Eßer 	bool comment = false, hash = false;
581252884aeSStefan Eßer 
582252884aeSStefan Eßer 	bc_lex_file(&vm.prs.l, bc_program_stdin_name);
583252884aeSStefan Eßer 
584252884aeSStefan Eßer 	BC_SIG_LOCK;
585252884aeSStefan Eßer 	bc_vec_init(&buffer, sizeof(uchar), NULL);
586252884aeSStefan Eßer 	bc_vec_init(&buf, sizeof(uchar), NULL);
587252884aeSStefan Eßer 	bc_vec_pushByte(&buffer, '\0');
588252884aeSStefan Eßer 	BC_SETJMP_LOCKED(err);
589252884aeSStefan Eßer 	BC_SIG_UNLOCK;
590252884aeSStefan Eßer 
591252884aeSStefan Eßer restart:
592252884aeSStefan Eßer 
593252884aeSStefan Eßer 	// This loop is complex because the vm tries not to send any lines that end
594252884aeSStefan Eßer 	// with a backslash to the parser. The reason for that is because the parser
595252884aeSStefan Eßer 	// treats a backslash+newline combo as whitespace, per the bc spec. In that
596252884aeSStefan Eßer 	// case, and for strings and comments, the parser will expect more stuff.
597252884aeSStefan Eßer 	while ((!(s = bc_read_line(&buf, ">>> ")) ||
598252884aeSStefan Eßer 	        (vm.eof = (s == BC_STATUS_EOF))) && buf.len > 1)
599252884aeSStefan Eßer 	{
600252884aeSStefan Eßer 		char c2, *str = buf.v;
601252884aeSStefan Eßer 		size_t i, len = buf.len - 1;
602252884aeSStefan Eßer 
603252884aeSStefan Eßer 		for (i = 0; i < len; ++i) {
604252884aeSStefan Eßer 
605252884aeSStefan Eßer 			bool notend = len > i + 1;
606252884aeSStefan Eßer 			uchar c = (uchar) str[i];
607252884aeSStefan Eßer 
608252884aeSStefan Eßer 			hash = (!comment && !string && ((hash && c != '\n') ||
609252884aeSStefan Eßer 			                                (!hash && c == '#')));
610252884aeSStefan Eßer 
611252884aeSStefan Eßer 			if (!hash && !comment && (i - 1 > len || str[i - 1] != '\\')) {
612252884aeSStefan Eßer 				if (BC_IS_BC) string ^= (c == '"');
613252884aeSStefan Eßer 				else if (c == ']') string -= 1;
614252884aeSStefan Eßer 				else if (c == '[') string += 1;
615252884aeSStefan Eßer 			}
616252884aeSStefan Eßer 
617252884aeSStefan Eßer 			if (BC_IS_BC && !hash && !string && notend) {
618252884aeSStefan Eßer 
619252884aeSStefan Eßer 				c2 = str[i + 1];
620252884aeSStefan Eßer 
621252884aeSStefan Eßer 				if (c == '/' && !comment && c2 == '*') {
622252884aeSStefan Eßer 					comment = true;
623252884aeSStefan Eßer 					i += 1;
624252884aeSStefan Eßer 				}
625252884aeSStefan Eßer 				else if (c == '*' && comment && c2 == '/') {
626252884aeSStefan Eßer 					comment = false;
627252884aeSStefan Eßer 					i += 1;
628252884aeSStefan Eßer 				}
629252884aeSStefan Eßer 			}
630252884aeSStefan Eßer 		}
631252884aeSStefan Eßer 
632252884aeSStefan Eßer 		bc_vec_concat(&buffer, buf.v);
633252884aeSStefan Eßer 
634252884aeSStefan Eßer 		if (string || comment) continue;
635252884aeSStefan Eßer 		if (len >= 2 && str[len - 2] == '\\' && str[len - 1] == '\n') continue;
636252884aeSStefan Eßer #if BC_ENABLE_HISTORY
637252884aeSStefan Eßer 		if (vm.history.stdin_has_data) continue;
638252884aeSStefan Eßer #endif // BC_ENABLE_HISTORY
639252884aeSStefan Eßer 
6405d934bc0SStefan Eßer 		bc_vm_process(buffer.v);
641252884aeSStefan Eßer 		bc_vec_empty(&buffer);
642252884aeSStefan Eßer 
643252884aeSStefan Eßer 		if (vm.eof) break;
6443aa99676SStefan Eßer 		else bc_vm_clean();
645252884aeSStefan Eßer 	}
646252884aeSStefan Eßer 
647252884aeSStefan Eßer 	if (!BC_STATUS_IS_ERROR(s)) {
648252884aeSStefan Eßer 		if (BC_ERR(comment))
649*50696a6eSStefan Eßer 			bc_parse_err(&vm.prs, BC_ERR_PARSE_COMMENT);
650252884aeSStefan Eßer 		else if (BC_ERR(string))
651*50696a6eSStefan Eßer 			bc_parse_err(&vm.prs, BC_ERR_PARSE_STRING);
652252884aeSStefan Eßer #if BC_ENABLED
6535d934bc0SStefan Eßer 		else if (BC_IS_BC) bc_vm_endif();
654252884aeSStefan Eßer #endif // BC_ENABLED
655252884aeSStefan Eßer 	}
656252884aeSStefan Eßer 
657252884aeSStefan Eßer err:
658252884aeSStefan Eßer 	BC_SIG_MAYLOCK;
659252884aeSStefan Eßer 
660252884aeSStefan Eßer 	bc_vm_clean();
661252884aeSStefan Eßer 
662252884aeSStefan Eßer 	vm.status = vm.status == BC_STATUS_ERROR_FATAL ||
663252884aeSStefan Eßer 	            vm.status == BC_STATUS_QUIT || !BC_I ?
664252884aeSStefan Eßer 	            vm.status : BC_STATUS_SUCCESS;
665252884aeSStefan Eßer 
666252884aeSStefan Eßer 	if (!vm.status && !vm.eof) {
667252884aeSStefan Eßer 		bc_vec_empty(&buffer);
668252884aeSStefan Eßer 		BC_LONGJMP_STOP;
669252884aeSStefan Eßer 		BC_SIG_UNLOCK;
670252884aeSStefan Eßer 		goto restart;
671252884aeSStefan Eßer 	}
672252884aeSStefan Eßer 
673252884aeSStefan Eßer 	bc_vec_free(&buf);
674252884aeSStefan Eßer 	bc_vec_free(&buffer);
675252884aeSStefan Eßer 
676252884aeSStefan Eßer 	BC_LONGJMP_CONT;
677252884aeSStefan Eßer }
678252884aeSStefan Eßer 
679252884aeSStefan Eßer #if BC_ENABLED
680252884aeSStefan Eßer static void bc_vm_load(const char *name, const char *text) {
681252884aeSStefan Eßer 
682252884aeSStefan Eßer 	bc_lex_file(&vm.prs.l, name);
683252884aeSStefan Eßer 	bc_parse_text(&vm.prs, text);
684252884aeSStefan Eßer 
685252884aeSStefan Eßer 	while (vm.prs.l.t != BC_LEX_EOF) vm.parse(&vm.prs);
686252884aeSStefan Eßer }
687252884aeSStefan Eßer #endif // BC_ENABLED
688252884aeSStefan Eßer 
689252884aeSStefan Eßer static void bc_vm_defaultMsgs(void) {
690252884aeSStefan Eßer 
691252884aeSStefan Eßer 	size_t i;
692252884aeSStefan Eßer 
693252884aeSStefan Eßer 	vm.func_header = bc_err_func_header;
694252884aeSStefan Eßer 
695252884aeSStefan Eßer 	for (i = 0; i < BC_ERR_IDX_NELEMS + BC_ENABLED; ++i)
696252884aeSStefan Eßer 		vm.err_ids[i] = bc_errs[i];
697*50696a6eSStefan Eßer 	for (i = 0; i < BC_ERR_NELEMS; ++i) vm.err_msgs[i] = bc_err_msgs[i];
698252884aeSStefan Eßer }
699252884aeSStefan Eßer 
700252884aeSStefan Eßer static void bc_vm_gettext(void) {
701252884aeSStefan Eßer 
702252884aeSStefan Eßer #if BC_ENABLE_NLS
703252884aeSStefan Eßer 	uchar id = 0;
704252884aeSStefan Eßer 	int set = 1, msg = 1;
705252884aeSStefan Eßer 	size_t i;
706252884aeSStefan Eßer 
707252884aeSStefan Eßer 	if (vm.locale == NULL) {
708252884aeSStefan Eßer 		vm.catalog = BC_VM_INVALID_CATALOG;
709252884aeSStefan Eßer 		bc_vm_defaultMsgs();
710252884aeSStefan Eßer 		return;
711252884aeSStefan Eßer 	}
712252884aeSStefan Eßer 
713252884aeSStefan Eßer 	vm.catalog = catopen(BC_MAINEXEC, NL_CAT_LOCALE);
714252884aeSStefan Eßer 
715252884aeSStefan Eßer 	if (vm.catalog == BC_VM_INVALID_CATALOG) {
716252884aeSStefan Eßer 		bc_vm_defaultMsgs();
717252884aeSStefan Eßer 		return;
718252884aeSStefan Eßer 	}
719252884aeSStefan Eßer 
720252884aeSStefan Eßer 	vm.func_header = catgets(vm.catalog, set, msg, bc_err_func_header);
721252884aeSStefan Eßer 
722252884aeSStefan Eßer 	for (set += 1; msg <= BC_ERR_IDX_NELEMS + BC_ENABLED; ++msg)
723252884aeSStefan Eßer 		vm.err_ids[msg - 1] = catgets(vm.catalog, set, msg, bc_errs[msg - 1]);
724252884aeSStefan Eßer 
725252884aeSStefan Eßer 	i = 0;
726252884aeSStefan Eßer 	id = bc_err_ids[i];
727252884aeSStefan Eßer 
728*50696a6eSStefan Eßer 	for (set = id + 3, msg = 1; i < BC_ERR_NELEMS; ++i, ++msg) {
729252884aeSStefan Eßer 
730252884aeSStefan Eßer 		if (id != bc_err_ids[i]) {
731252884aeSStefan Eßer 			msg = 1;
732252884aeSStefan Eßer 			id = bc_err_ids[i];
733252884aeSStefan Eßer 			set = id + 3;
734252884aeSStefan Eßer 		}
735252884aeSStefan Eßer 
736252884aeSStefan Eßer 		vm.err_msgs[i] = catgets(vm.catalog, set, msg, bc_err_msgs[i]);
737252884aeSStefan Eßer 	}
738252884aeSStefan Eßer #else // BC_ENABLE_NLS
739252884aeSStefan Eßer 	bc_vm_defaultMsgs();
740252884aeSStefan Eßer #endif // BC_ENABLE_NLS
741252884aeSStefan Eßer }
742252884aeSStefan Eßer 
7435d934bc0SStefan Eßer static void bc_vm_exec(void) {
744252884aeSStefan Eßer 
745252884aeSStefan Eßer 	size_t i;
746252884aeSStefan Eßer 	bool has_file = false;
7473aa99676SStefan Eßer 	BcVec buf;
748252884aeSStefan Eßer 
749252884aeSStefan Eßer #if BC_ENABLED
750252884aeSStefan Eßer 	if (BC_IS_BC && (vm.flags & BC_FLAG_L)) {
751252884aeSStefan Eßer 
752252884aeSStefan Eßer 		bc_vm_load(bc_lib_name, bc_lib);
753252884aeSStefan Eßer 
754252884aeSStefan Eßer #if BC_ENABLE_EXTRA_MATH
755252884aeSStefan Eßer 		if (!BC_IS_POSIX) bc_vm_load(bc_lib2_name, bc_lib2);
756252884aeSStefan Eßer #endif // BC_ENABLE_EXTRA_MATH
757252884aeSStefan Eßer 	}
758252884aeSStefan Eßer #endif // BC_ENABLED
759252884aeSStefan Eßer 
760252884aeSStefan Eßer 	if (vm.exprs.len) {
7613aa99676SStefan Eßer 
7623aa99676SStefan Eßer 		size_t len = vm.exprs.len - 1;
7633aa99676SStefan Eßer 		bool more;
7643aa99676SStefan Eßer 
7653aa99676SStefan Eßer 		BC_SIG_LOCK;
7663aa99676SStefan Eßer 		bc_vec_init(&buf, sizeof(uchar), NULL);
7673aa99676SStefan Eßer 
7683aa99676SStefan Eßer #ifndef NDEBUG
7693aa99676SStefan Eßer 		BC_SETJMP_LOCKED(err);
7703aa99676SStefan Eßer #endif // NDEBUG
7713aa99676SStefan Eßer 
7723aa99676SStefan Eßer 		BC_SIG_UNLOCK;
7733aa99676SStefan Eßer 
774252884aeSStefan Eßer 		bc_lex_file(&vm.prs.l, bc_program_exprs_name);
7753aa99676SStefan Eßer 
7763aa99676SStefan Eßer 		do {
7773aa99676SStefan Eßer 
7783aa99676SStefan Eßer 			more = bc_read_buf(&buf, vm.exprs.v, &len);
7793aa99676SStefan Eßer 			bc_vec_pushByte(&buf, '\0');
7805d934bc0SStefan Eßer 			bc_vm_process(buf.v);
7813aa99676SStefan Eßer 
7823aa99676SStefan Eßer 			bc_vec_npop(&buf, buf.len);
7833aa99676SStefan Eßer 
7843aa99676SStefan Eßer 		} while (more);
7853aa99676SStefan Eßer 
7863aa99676SStefan Eßer 		BC_SIG_LOCK;
7873aa99676SStefan Eßer 		bc_vec_free(&buf);
7883aa99676SStefan Eßer 
7893aa99676SStefan Eßer #ifndef NDEBUG
7903aa99676SStefan Eßer 		BC_UNSETJMP;
7913aa99676SStefan Eßer #endif // NDEBUG
7923aa99676SStefan Eßer 
7933aa99676SStefan Eßer 		BC_SIG_UNLOCK;
7943aa99676SStefan Eßer 
7955d934bc0SStefan Eßer 		if (!vm.no_exit_exprs) return;
796252884aeSStefan Eßer 	}
797252884aeSStefan Eßer 
798252884aeSStefan Eßer 	for (i = 0; i < vm.files.len; ++i) {
799252884aeSStefan Eßer 		char *path = *((char**) bc_vec_item(&vm.files, i));
800252884aeSStefan Eßer 		if (!strcmp(path, "")) continue;
801252884aeSStefan Eßer 		has_file = true;
802252884aeSStefan Eßer 		bc_vm_file(path);
803252884aeSStefan Eßer 	}
804252884aeSStefan Eßer 
805252884aeSStefan Eßer 	if (BC_IS_BC || !has_file) bc_vm_stdin();
8063aa99676SStefan Eßer 
8073aa99676SStefan Eßer // These are all protected by ifndef NDEBUG because if these are needed, bc is
8083aa99676SStefan Eßer // goingi to exit anyway, and I see no reason to include this code in a release
8093aa99676SStefan Eßer // build when the OS is going to free all of the resources anyway.
8103aa99676SStefan Eßer #ifndef NDEBUG
8113aa99676SStefan Eßer 	return;
8123aa99676SStefan Eßer 
8133aa99676SStefan Eßer err:
8143aa99676SStefan Eßer 	BC_SIG_MAYLOCK;
8153aa99676SStefan Eßer 	bc_vec_free(&buf);
8163aa99676SStefan Eßer 	BC_LONGJMP_CONT;
8173aa99676SStefan Eßer #endif // NDEBUG
818252884aeSStefan Eßer }
819252884aeSStefan Eßer 
820252884aeSStefan Eßer void bc_vm_boot(int argc, char *argv[], const char *env_len,
8215d934bc0SStefan Eßer                 const char* const env_args)
822252884aeSStefan Eßer {
823252884aeSStefan Eßer 	int ttyin, ttyout, ttyerr;
824252884aeSStefan Eßer 	struct sigaction sa;
825252884aeSStefan Eßer 
826252884aeSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
827252884aeSStefan Eßer 
828252884aeSStefan Eßer 	ttyin = isatty(STDIN_FILENO);
829252884aeSStefan Eßer 	ttyout = isatty(STDOUT_FILENO);
830252884aeSStefan Eßer 	ttyerr = isatty(STDERR_FILENO);
831252884aeSStefan Eßer 
832252884aeSStefan Eßer 	vm.flags |= ttyin ? BC_FLAG_TTYIN : 0;
833252884aeSStefan Eßer 	vm.flags |= (ttyin != 0 && ttyout != 0 && ttyerr != 0) ? BC_FLAG_TTY : 0;
834252884aeSStefan Eßer 	vm.flags |= ttyin && ttyout ? BC_FLAG_I : 0;
835252884aeSStefan Eßer 
836252884aeSStefan Eßer 	sigemptyset(&sa.sa_mask);
837252884aeSStefan Eßer 	sa.sa_handler = bc_vm_sig;
838252884aeSStefan Eßer 	sa.sa_flags = SA_NODEFER;
839252884aeSStefan Eßer 
840252884aeSStefan Eßer 	sigaction(SIGTERM, &sa, NULL);
841252884aeSStefan Eßer 	sigaction(SIGQUIT, &sa, NULL);
842252884aeSStefan Eßer 	sigaction(SIGINT, &sa, NULL);
843252884aeSStefan Eßer 
844252884aeSStefan Eßer #if BC_ENABLE_HISTORY
845252884aeSStefan Eßer 	if (BC_TTY) sigaction(SIGHUP, &sa, NULL);
846252884aeSStefan Eßer #endif // BC_ENABLE_HISTORY
847252884aeSStefan Eßer 
848*50696a6eSStefan Eßer 	bc_vm_init();
849252884aeSStefan Eßer 
850252884aeSStefan Eßer 	vm.file = NULL;
851252884aeSStefan Eßer 
852252884aeSStefan Eßer 	bc_vm_gettext();
853252884aeSStefan Eßer 
854252884aeSStefan Eßer 	bc_file_init(&vm.ferr, STDERR_FILENO, output_bufs + BC_VM_STDOUT_BUF_SIZE,
855252884aeSStefan Eßer 	             BC_VM_STDERR_BUF_SIZE);
856252884aeSStefan Eßer 	bc_file_init(&vm.fout, STDOUT_FILENO, output_bufs, BC_VM_STDOUT_BUF_SIZE);
857252884aeSStefan Eßer 	vm.buf = output_bufs + BC_VM_STDOUT_BUF_SIZE + BC_VM_STDERR_BUF_SIZE;
858252884aeSStefan Eßer 
859252884aeSStefan Eßer 	vm.line_len = (uint16_t) bc_vm_envLen(env_len);
860252884aeSStefan Eßer 
861252884aeSStefan Eßer 	bc_vec_clear(&vm.files);
862252884aeSStefan Eßer 	bc_vec_clear(&vm.exprs);
863252884aeSStefan Eßer 
864252884aeSStefan Eßer 	bc_program_init(&vm.prog);
865252884aeSStefan Eßer 	bc_parse_init(&vm.prs, &vm.prog, BC_PROG_MAIN);
866252884aeSStefan Eßer 
867252884aeSStefan Eßer #if BC_ENABLE_HISTORY
868252884aeSStefan Eßer 	if (BC_TTY) bc_history_init(&vm.history);
869252884aeSStefan Eßer #endif // BC_ENABLE_HISTORY
870252884aeSStefan Eßer 
871252884aeSStefan Eßer #if BC_ENABLED
872252884aeSStefan Eßer 	if (BC_IS_BC) vm.flags |= BC_FLAG_S * (getenv("POSIXLY_CORRECT") != NULL);
873252884aeSStefan Eßer #endif // BC_ENABLED
874252884aeSStefan Eßer 
875252884aeSStefan Eßer 	bc_vm_envArgs(env_args);
876252884aeSStefan Eßer 	bc_args(argc, argv);
877252884aeSStefan Eßer 
8783aa99676SStefan Eßer #if BC_ENABLED
879252884aeSStefan Eßer 	if (BC_IS_POSIX) vm.flags &= ~(BC_FLAG_G);
8803aa99676SStefan Eßer #endif // BC_ENABLED
881252884aeSStefan Eßer 
882*50696a6eSStefan Eßer 	BC_SIG_UNLOCK;
883*50696a6eSStefan Eßer 
884*50696a6eSStefan Eßer 	bc_vm_exec();
885*50696a6eSStefan Eßer }
886*50696a6eSStefan Eßer #endif // !BC_ENABLE_LIBRARY
887*50696a6eSStefan Eßer 
888*50696a6eSStefan Eßer void bc_vm_init(void) {
889*50696a6eSStefan Eßer 
890*50696a6eSStefan Eßer 	BC_SIG_ASSERT_LOCKED;
891*50696a6eSStefan Eßer 
892*50696a6eSStefan Eßer 	memcpy(vm.max_num, bc_num_bigdigMax,
893*50696a6eSStefan Eßer 	       bc_num_bigdigMax_size * sizeof(BcDig));
894*50696a6eSStefan Eßer 	memcpy(vm.max2_num, bc_num_bigdigMax2,
895*50696a6eSStefan Eßer 	       bc_num_bigdigMax2_size * sizeof(BcDig));
896*50696a6eSStefan Eßer 	bc_num_setup(&vm.max, vm.max_num, BC_NUM_BIGDIG_LOG10);
897*50696a6eSStefan Eßer 	bc_num_setup(&vm.max2, vm.max2_num, BC_NUM_BIGDIG_LOG10);
898*50696a6eSStefan Eßer 	vm.max.len = bc_num_bigdigMax_size;
899*50696a6eSStefan Eßer 	vm.max2.len = bc_num_bigdigMax2_size;
900*50696a6eSStefan Eßer 
901*50696a6eSStefan Eßer 	bc_vec_init(&vm.temps, sizeof(BcNum), NULL);
902*50696a6eSStefan Eßer 
903252884aeSStefan Eßer 	vm.maxes[BC_PROG_GLOBALS_IBASE] = BC_NUM_MAX_POSIX_IBASE;
904252884aeSStefan Eßer 	vm.maxes[BC_PROG_GLOBALS_OBASE] = BC_MAX_OBASE;
905252884aeSStefan Eßer 	vm.maxes[BC_PROG_GLOBALS_SCALE] = BC_MAX_SCALE;
906252884aeSStefan Eßer 
9073aa99676SStefan Eßer #if BC_ENABLE_EXTRA_MATH && BC_ENABLE_RAND
908252884aeSStefan Eßer 	vm.maxes[BC_PROG_MAX_RAND] = ((BcRand) 0) - 1;
9093aa99676SStefan Eßer #endif // BC_ENABLE_EXTRA_MATH && BC_ENABLE_RAND
910252884aeSStefan Eßer 
9113aa99676SStefan Eßer #if BC_ENABLED
912*50696a6eSStefan Eßer #if !BC_ENABLE_LIBRARY
913252884aeSStefan Eßer 	if (BC_IS_BC && !BC_IS_POSIX)
914*50696a6eSStefan Eßer #endif // !BC_ENABLE_LIBRARY
915*50696a6eSStefan Eßer 	{
916252884aeSStefan Eßer 		vm.maxes[BC_PROG_GLOBALS_IBASE] = BC_NUM_MAX_IBASE;
917*50696a6eSStefan Eßer 	}
9183aa99676SStefan Eßer #endif // BC_ENABLED
919252884aeSStefan Eßer }
920