#include <assert.h>
#include <ctype.h>
#include <errno.h>
#include <stdarg.h>
#include <string.h>
#include <signal.h>
#include <setjmp.h>
#ifndef _WIN32
#include <unistd.h>
#include <sys/types.h>
#include <unistd.h>
#else
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <io.h>
#endif
#include <status.h>
#include <vector.h>
#include <args.h>
#include <vm.h>
#include <read.h>
#include <bc.h>
#if BC_ENABLE_LIBRARY
#include <library.h>
#endif
#if BC_ENABLE_OSSFUZZ
#include <ossfuzz.h>
#endif
#if !BC_ENABLE_LIBRARY
char output_bufs[BC_VM_BUF_SIZE];
BcVm vm_data;
BcVm* vm = &vm_data;
#endif
#if BC_DEBUG_CODE
BC_NORETURN void
bc_vm_jmp(const char* f)
{
#else
BC_NORETURN void
bc_vm_jmp(void)
{
#endif
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
assert(BC_SIG_EXC(vm));
BC_SIG_MAYLOCK;
#if BC_DEBUG_CODE
bc_file_puts(&vm->ferr, bc_flush_none, "Longjmp: ");
bc_file_puts(&vm->ferr, bc_flush_none, f);
bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
bc_file_flush(&vm->ferr, bc_flush_none);
#endif
#if BC_DEBUG
assert(vm->jmp_bufs.len - (size_t) vm->sig_pop);
#endif
if (vm->jmp_bufs.len == 0) abort();
if (vm->sig_pop) bc_vec_pop(&vm->jmp_bufs);
else vm->sig_pop = 1;
siglongjmp(*((sigjmp_buf*) bc_vec_top(&vm->jmp_bufs)), 1);
}
#if !BC_ENABLE_LIBRARY
static void
bc_vm_sig(int sig)
{
if (vm->status == (sig_atomic_t) BC_STATUS_QUIT || vm->sig != 0)
{
if (!BC_I || sig != SIGINT) vm->status = BC_STATUS_QUIT;
return;
}
vm->sig = sig;
if (sig == SIGINT && BC_SIGINT && BC_I)
{
int err = errno;
#if BC_ENABLE_EDITLINE
if (write(STDOUT_FILENO, "^C", 2) != (ssize_t) 2)
{
vm->status = BC_STATUS_ERROR_FATAL;
}
#endif
if (write(STDOUT_FILENO, vm->sigmsg, vm->siglen) !=
(ssize_t) vm->siglen)
{
vm->status = BC_STATUS_ERROR_FATAL;
}
errno = err;
}
else
{
#if BC_ENABLE_EDITLINE
if (write(STDOUT_FILENO, "^C", 2) != (ssize_t) 2)
{
vm->status = BC_STATUS_ERROR_FATAL;
return;
}
#endif
vm->status = BC_STATUS_QUIT;
}
#if BC_ENABLE_LINE_LIB
if (sig == SIGINT && bc_history_inlinelib)
{
bc_history_inlinelib = 0;
siglongjmp(bc_history_jmpbuf, 1);
}
#endif
assert(vm->jmp_bufs.len);
if (!vm->sig_lock) BC_JMP;
}
static void
bc_vm_sigaction(void)
{
#ifndef _WIN32
struct sigaction sa;
sigemptyset(&sa.sa_mask);
sa.sa_flags = SA_NODEFER;
#if BC_CLANG
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wdisabled-macro-expansion"
#endif
sa.sa_handler = bc_vm_sig;
#if BC_CLANG
#pragma clang diagnostic pop
#endif
sigaction(SIGTERM, &sa, NULL);
sigaction(SIGQUIT, &sa, NULL);
sigaction(SIGINT, &sa, NULL);
#if BC_ENABLE_HISTORY
if (BC_TTY) sigaction(SIGHUP, &sa, NULL);
#endif
#else
signal(SIGTERM, bc_vm_sig);
signal(SIGINT, bc_vm_sig);
#endif
}
void
bc_vm_info(const char* const help)
{
BC_SIG_ASSERT_LOCKED;
bc_file_printf(&vm->fout, "%s %s\n%s", vm->name, BC_VERSION, bc_copyright);
if (help != NULL)
{
bc_file_putchar(&vm->fout, bc_flush_none, '\n');
#if BC_ENABLED
if (BC_IS_BC)
{
const char* const banner = BC_DEFAULT_BANNER ? "to" : "to not";
const char* const sigint = BC_DEFAULT_SIGINT_RESET ? "to reset" :
"to exit";
const char* const tty = BC_DEFAULT_TTY_MODE ? "enabled" :
"disabled";
const char* const prompt = BC_DEFAULT_PROMPT ? "enabled" :
"disabled";
const char* const expr = BC_DEFAULT_EXPR_EXIT ? "to exit" :
"to not exit";
const char* const clamp = BC_DEFAULT_DIGIT_CLAMP ? "to clamp" :
"to not clamp";
bc_file_printf(&vm->fout, help, vm->name, vm->name, BC_VERSION,
BC_BUILD_TYPE, banner, sigint, tty, prompt, expr,
clamp);
}
#endif
#if DC_ENABLED
if (BC_IS_DC)
{
const char* const sigint = DC_DEFAULT_SIGINT_RESET ? "to reset" :
"to exit";
const char* const tty = DC_DEFAULT_TTY_MODE ? "enabled" :
"disabled";
const char* const prompt = DC_DEFAULT_PROMPT ? "enabled" :
"disabled";
const char* const expr = DC_DEFAULT_EXPR_EXIT ? "to exit" :
"to not exit";
const char* const clamp = DC_DEFAULT_DIGIT_CLAMP ? "to clamp" :
"to not clamp";
bc_file_printf(&vm->fout, help, vm->name, vm->name, BC_VERSION,
BC_BUILD_TYPE, sigint, tty, prompt, expr, clamp);
}
#endif
}
bc_file_flush(&vm->fout, bc_flush_none);
}
#endif
#if !BC_ENABLE_LIBRARY && !BC_ENABLE_MEMCHECK
BC_NORETURN
#endif
void
bc_vm_fatalError(BcErr e)
{
bc_err(e);
#if !BC_ENABLE_LIBRARY && !BC_ENABLE_MEMCHECK
BC_UNREACHABLE
#if !BC_CLANG
abort();
#endif
#endif
}
#if BC_ENABLE_LIBRARY
BC_NORETURN void
bc_vm_handleError(BcErr e)
{
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
assert(e < BC_ERR_NELEMS);
assert(!vm->sig_pop);
BC_SIG_LOCK;
if (e <= BC_ERR_MATH_DIVIDE_BY_ZERO)
{
vm->err = (BclError) (e - BC_ERR_MATH_NEGATIVE +
BCL_ERROR_MATH_NEGATIVE);
}
else if (vm->abrt) abort();
else if (e == BC_ERR_FATAL_ALLOC_ERR) vm->err = BCL_ERROR_FATAL_ALLOC_ERR;
else vm->err = BCL_ERROR_FATAL_UNKNOWN_ERR;
BC_JMP;
}
#else
#if BC_DEBUG
void
bc_vm_handleError(BcErr e, const char* file, int fline, size_t line, ...)
#else
void
bc_vm_handleError(BcErr e, size_t line, ...)
#endif
{
BcStatus s;
BcStatus fout_s;
va_list args;
uchar id = bc_err_ids[e];
const char* err_type = vm->err_ids[id];
sig_atomic_t lock;
assert(e < BC_ERR_NELEMS);
assert(!vm->sig_pop);
#if BC_ENABLED
if (!BC_S && e >= BC_ERR_POSIX_START)
{
if (BC_W)
{
id = UCHAR_MAX;
err_type = vm->err_ids[BC_ERR_IDX_WARN];
}
else return;
}
#endif
BC_SIG_TRYLOCK(lock);
fout_s = bc_file_flushErr(&vm->fout, bc_flush_err);
va_start(args, line);
bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
bc_file_puts(&vm->ferr, bc_flush_none, err_type);
bc_file_putchar(&vm->ferr, bc_flush_none, ' ');
bc_file_vprintf(&vm->ferr, vm->err_msgs[e], args);
va_end(args);
if (BC_NO_ERR(vm->file != NULL))
{
if (line)
{
bc_file_puts(&vm->ferr, bc_flush_none, "\n ");
bc_file_puts(&vm->ferr, bc_flush_none, vm->file);
bc_file_printf(&vm->ferr, ":%zu\n", line);
}
else
{
bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
bc_program_printStackTrace(&vm->prog);
}
}
else
{
bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
}
#if BC_DEBUG
bc_file_printf(&vm->ferr, "\n %s:%d\n", file, fline);
#endif
bc_file_puts(&vm->ferr, bc_flush_none, "\n");
if (fout_s == BC_STATUS_ERROR_FATAL && e != BC_ERR_FATAL_IO_ERR)
{
bc_file_putchar(&vm->ferr, bc_flush_none, '\n');
bc_file_puts(&vm->ferr, bc_flush_none,
vm->err_ids[bc_err_ids[BC_ERR_FATAL_IO_ERR]]);
bc_file_putchar(&vm->ferr, bc_flush_none, ' ');
bc_file_puts(&vm->ferr, bc_flush_none,
vm->err_msgs[BC_ERR_FATAL_IO_ERR]);
}
s = bc_file_flushErr(&vm->ferr, bc_flush_err);
#if !BC_ENABLE_MEMCHECK
if (BC_ERR(id == BC_ERR_IDX_FATAL || fout_s == BC_STATUS_ERROR_FATAL ||
s == BC_STATUS_ERROR_FATAL))
{
exit((int) BC_STATUS_ERROR_FATAL);
}
#else
if (BC_ERR(fout_s == BC_STATUS_ERROR_FATAL))
{
vm->status = (sig_atomic_t) fout_s;
}
else if (BC_ERR(s == BC_STATUS_ERROR_FATAL))
{
vm->status = (sig_atomic_t) s;
}
else
#endif
{
vm->status = (sig_atomic_t) (uchar) (id + 1);
}
if (BC_ERR(vm->status)) BC_JMP;
BC_SIG_TRYUNLOCK(lock);
}
char*
bc_vm_getenv(const char* var)
{
char* ret;
#ifndef _WIN32
ret = getenv(var);
#else
_dupenv_s(&ret, NULL, var);
#endif
return ret;
}
void
bc_vm_getenvFree(char* val)
{
BC_UNUSED(val);
#ifdef _WIN32
free(val);
#endif
}
static void
bc_vm_setenvFlag(const char* const var, int def, uint16_t flag)
{
char* val = bc_vm_getenv(var);
if (val == NULL)
{
if (def) vm->flags |= flag;
else vm->flags &= ~(flag);
}
else if (strtoul(val, NULL, 0)) vm->flags |= flag;
else vm->flags &= ~(flag);
bc_vm_getenvFree(val);
}
static void
bc_vm_envArgs(const char* const env_args_name, BcBigDig* scale, BcBigDig* ibase,
BcBigDig* obase)
{
char *env_args = bc_vm_getenv(env_args_name), *buf, *start;
char instr = '\0';
BC_SIG_ASSERT_LOCKED;
if (env_args == NULL) return;
#ifndef _WIN32
start = buf = vm->env_args_buffer = bc_vm_strdup(env_args);
#else
start = buf = vm->env_args_buffer = env_args;
#endif
assert(buf != NULL);
bc_vec_init(&vm->env_args, sizeof(char*), BC_DTOR_NONE);
bc_vec_push(&vm->env_args, &env_args_name);
while (*buf)
{
if (!isspace(*buf))
{
if (*buf == '"' || *buf == '\'')
{
instr = *buf;
buf += 1;
if (*buf == instr)
{
instr = '\0';
buf += 1;
continue;
}
}
bc_vec_push(&vm->env_args, &buf);
while (*buf &&
((!instr && !isspace(*buf)) || (instr && *buf != instr)))
{
buf += 1;
}
if (*buf)
{
if (instr) instr = '\0';
*buf = '\0';
buf += 1;
start = buf;
}
else if (instr) bc_error(BC_ERR_FATAL_OPTION, 0, start);
}
else buf += 1;
}
buf = NULL;
bc_vec_push(&vm->env_args, &buf);
bc_args((int) vm->env_args.len - 1, bc_vec_item(&vm->env_args, 0), false,
scale, ibase, obase);
}
static size_t
bc_vm_envLen(const char* var)
{
char* lenv = bc_vm_getenv(var);
size_t i, len = BC_NUM_PRINT_WIDTH;
int num;
if (lenv == NULL) return len;
len = strlen(lenv);
for (num = 1, i = 0; num && i < len; ++i)
{
num = isdigit(lenv[i]);
}
if (num)
{
len = (size_t) strtol(lenv, NULL, 10);
if (len != 0)
{
len -= 1;
if (len < 2 || len >= UINT16_MAX) len = BC_NUM_PRINT_WIDTH;
}
}
else len = BC_NUM_PRINT_WIDTH;
bc_vm_getenvFree(lenv);
return len;
}
#endif
void
bc_vm_shutdown(void)
{
BC_SIG_ASSERT_LOCKED;
#if BC_ENABLE_NLS
if (vm->catalog != BC_VM_INVALID_CATALOG) catclose(vm->catalog);
#endif
#if !BC_ENABLE_LIBRARY
#if BC_ENABLE_HISTORY
if (BC_TTY && !vm->history.badTerm) bc_history_free(&vm->history);
#endif
#endif
#if BC_DEBUG || BC_ENABLE_MEMCHECK
#if !BC_ENABLE_LIBRARY
bc_vec_free(&vm->env_args);
free(vm->env_args_buffer);
bc_vec_free(&vm->files);
bc_vec_free(&vm->exprs);
if (BC_PARSE_IS_INITED(&vm->read_prs, &vm->prog))
{
bc_vec_free(&vm->read_buf);
bc_parse_free(&vm->read_prs);
}
bc_parse_free(&vm->prs);
bc_program_free(&vm->prog);
bc_slabvec_free(&vm->slabs);
#endif
bc_vm_freeTemps();
#endif
#if !BC_ENABLE_LIBRARY
bc_file_free(&vm->fout);
bc_file_free(&vm->ferr);
#endif
}
void
bc_vm_addTemp(BcDig* num)
{
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
BC_SIG_ASSERT_LOCKED;
if (vm->temps_len == BC_VM_MAX_TEMPS) free(num);
else
{
vm->temps_buf[vm->temps_len] = num;
vm->temps_len += 1;
}
}
BcDig*
bc_vm_takeTemp(void)
{
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
BC_SIG_ASSERT_LOCKED;
if (!vm->temps_len) return NULL;
vm->temps_len -= 1;
return vm->temps_buf[vm->temps_len];
}
BcDig*
bc_vm_getTemp(void)
{
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
BC_SIG_ASSERT_LOCKED;
if (!vm->temps_len) return NULL;
return vm->temps_buf[vm->temps_len - 1];
}
void
bc_vm_freeTemps(void)
{
size_t i;
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
BC_SIG_ASSERT_LOCKED;
if (!vm->temps_len) return;
for (i = 0; i < vm->temps_len; ++i)
{
free(vm->temps_buf[i]);
}
vm->temps_len = 0;
}
#if !BC_ENABLE_LIBRARY
size_t
bc_vm_numDigits(size_t val)
{
size_t digits = 0;
do
{
digits += 1;
val /= 10;
}
while (val != 0);
return digits;
}
#endif
inline size_t
bc_vm_arraySize(size_t n, size_t size)
{
size_t res = n * size;
if (BC_ERR(BC_VM_MUL_OVERFLOW(n, size, res)))
{
bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
}
return res;
}
inline size_t
bc_vm_growSize(size_t a, size_t b)
{
size_t res = a + b;
if (BC_ERR(res >= SIZE_MAX || res < a))
{
bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
}
return res;
}
void*
bc_vm_malloc(size_t n)
{
void* ptr;
BC_SIG_ASSERT_LOCKED;
ptr = malloc(n);
if (BC_ERR(ptr == NULL))
{
bc_vm_freeTemps();
ptr = malloc(n);
if (BC_ERR(ptr == NULL)) bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
}
return ptr;
}
void*
bc_vm_realloc(void* ptr, size_t n)
{
void* temp;
BC_SIG_ASSERT_LOCKED;
temp = realloc(ptr, n);
if (BC_ERR(temp == NULL))
{
bc_vm_freeTemps();
temp = realloc(ptr, n);
if (BC_ERR(temp == NULL)) bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
}
return temp;
}
char*
bc_vm_strdup(const char* str)
{
char* s;
BC_SIG_ASSERT_LOCKED;
s = strdup(str);
if (BC_ERR(s == NULL))
{
bc_vm_freeTemps();
s = strdup(str);
if (BC_ERR(s == NULL)) bc_vm_fatalError(BC_ERR_FATAL_ALLOC_ERR);
}
return s;
}
#if !BC_ENABLE_LIBRARY
void
bc_vm_printf(const char* fmt, ...)
{
va_list args;
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#else
sig_atomic_t lock;
#endif
BC_SIG_TRYLOCK(lock);
va_start(args, fmt);
bc_file_vprintf(&vm->fout, fmt, args);
va_end(args);
vm->nchars = 0;
BC_SIG_TRYUNLOCK(lock);
}
#endif
void
bc_vm_putchar(int c, BcFlushType type)
{
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
bc_vec_pushByte(&vm->out, (uchar) c);
#else
bc_file_putchar(&vm->fout, type, (uchar) c);
vm->nchars = (c == '\n' ? 0 : vm->nchars + 1);
#endif
}
#if !BC_ENABLE_LIBRARY
#ifdef __OpenBSD__
BC_NORETURN static void
bc_abortm(const char* msg)
{
bc_file_puts(&vm->ferr, bc_flush_none, msg);
bc_file_puts(&vm->ferr, bc_flush_none, "; this is a bug");
bc_file_flush(&vm->ferr, bc_flush_none);
abort();
}
void
bc_pledge(const char* promises, const char* execpromises)
{
int r = pledge(promises, execpromises);
if (r) bc_abortm("pledge() failed");
}
#if BC_ENABLE_EXTRA_MATH
static void
bc_unveil(const char* path, const char* permissions)
{
int r = unveil(path, permissions);
if (r) bc_abortm("unveil() failed");
}
#endif
#else
void
bc_pledge(const char* promises, const char* execpromises)
{
BC_UNUSED(promises);
BC_UNUSED(execpromises);
}
#if BC_ENABLE_EXTRA_MATH
static void
bc_unveil(const char* path, const char* permissions)
{
BC_UNUSED(path);
BC_UNUSED(permissions);
}
#endif
#endif
static void
bc_vm_clean(void)
{
BcVec* fns = &vm->prog.fns;
BcFunc* f = bc_vec_item(fns, BC_PROG_MAIN);
BcInstPtr* ip = bc_vec_item(&vm->prog.stack, 0);
bool good = ((vm->status && vm->status != BC_STATUS_QUIT) || vm->sig != 0);
BC_SIG_ASSERT_LOCKED;
if (good) bc_program_reset(&vm->prog);
#if BC_ENABLED
if (good && BC_IS_BC) good = !BC_PARSE_NO_EXEC(&vm->prs);
#endif
#if DC_ENABLED
if (BC_IS_DC)
{
size_t i;
good = true;
for (i = 0; good && i < vm->prog.results.len; ++i)
{
BcResult* r = (BcResult*) bc_vec_item(&vm->prog.results, i);
good = BC_VM_SAFE_RESULT(r);
}
}
#endif
if (good && vm->prog.stack.len == 1 && ip->idx == f->code.len)
{
#if BC_ENABLED
if (BC_IS_BC)
{
bc_vec_popAll(&f->labels);
}
#endif
bc_vec_popAll(&f->code);
ip->idx = 0;
}
}
static void
bc_vm_process(const char* text, BcMode mode)
{
bc_parse_text(&vm->prs, text, mode);
while (vm->prs.l.t != BC_LEX_EOF)
{
BC_SIG_LOCK;
vm->parse(&vm->prs);
BC_SIG_UNLOCK;
if (BC_IS_DC || !BC_PARSE_NO_EXEC(&vm->prs)) bc_program_exec(&vm->prog);
assert(BC_IS_DC || vm->prog.results.len == 0);
if (BC_I) bc_file_flush(&vm->fout, bc_flush_save);
}
}
#if BC_ENABLED
static void
bc_vm_endif(void)
{
bc_parse_endif(&vm->prs);
bc_program_exec(&vm->prog);
}
#endif
static void
bc_vm_file(const char* file)
{
char* data = NULL;
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
assert(!vm->sig_pop);
vm->mode = BC_MODE_FILE;
bc_lex_file(&vm->prs.l, file);
BC_SIG_LOCK;
data = bc_read_file(file);
assert(data != NULL);
BC_SETJMP_LOCKED(vm, err);
BC_SIG_UNLOCK;
bc_vm_process(data, BC_MODE_FILE);
#if BC_ENABLED
if (BC_IS_BC) bc_vm_endif();
#endif
err:
BC_SIG_MAYLOCK;
free(data);
bc_vm_clean();
if (vm->status == (sig_atomic_t) BC_STATUS_SUCCESS) BC_LONGJMP_STOP;
BC_LONGJMP_CONT(vm);
}
#if !BC_ENABLE_OSSFUZZ
bool
bc_vm_readLine(bool clear)
{
BcStatus s;
bool good;
BC_SIG_ASSERT_NOT_LOCKED;
if (clear) bc_vec_empty(&vm->buffer);
bc_vec_empty(&vm->line_buf);
if (vm->eof) return false;
do
{
s = bc_read_line(&vm->line_buf, ">>> ");
vm->eof = (s == BC_STATUS_EOF);
}
while (s == BC_STATUS_SUCCESS && !vm->eof && vm->line_buf.len < 1);
good = (vm->line_buf.len > 1);
if (good) bc_vec_concat(&vm->buffer, vm->line_buf.v);
return good;
}
static void
bc_vm_stdin(void)
{
bool clear;
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
clear = true;
vm->mode = BC_MODE_STDIN;
bc_lex_file(&vm->prs.l, bc_program_stdin_name);
BC_SIG_LOCK;
bc_vec_init(&vm->buffer, sizeof(uchar), BC_DTOR_NONE);
bc_vec_init(&vm->line_buf, sizeof(uchar), BC_DTOR_NONE);
BC_SETJMP_LOCKED(vm, err);
BC_SIG_UNLOCK;
restart:
while (bc_vm_readLine(clear))
{
size_t len = vm->buffer.len - 1;
const char* str = vm->buffer.v;
clear = (len < 2 || str[len - 2] != '\\' || str[len - 1] != '\n');
if (!clear) continue;
bc_vm_process(vm->buffer.v, BC_MODE_STDIN);
if (vm->eof) break;
else
{
BC_SIG_LOCK;
bc_vm_clean();
BC_SIG_UNLOCK;
}
}
#if BC_ENABLED
if (BC_IS_BC) bc_vm_endif();
#endif
err:
BC_SIG_MAYLOCK;
bc_vm_clean();
#if !BC_ENABLE_MEMCHECK
assert(vm->status != BC_STATUS_ERROR_FATAL);
vm->status = vm->status == BC_STATUS_QUIT || !BC_I ? vm->status :
BC_STATUS_SUCCESS;
#else
vm->status = vm->status == BC_STATUS_ERROR_FATAL ||
vm->status == BC_STATUS_QUIT || !BC_I ?
vm->status :
BC_STATUS_SUCCESS;
#endif
if (!vm->status && !vm->eof)
{
bc_vec_empty(&vm->buffer);
BC_LONGJMP_STOP;
BC_SIG_UNLOCK;
goto restart;
}
#if BC_DEBUG
bc_vec_free(&vm->line_buf);
bc_vec_free(&vm->buffer);
#endif
BC_LONGJMP_CONT(vm);
}
#endif
bool
bc_vm_readBuf(bool clear)
{
size_t len = vm->exprs.len - 1;
bool more;
BC_SIG_ASSERT_NOT_LOCKED;
if (clear) bc_vec_empty(&vm->buffer);
bc_vec_pop(&vm->buffer);
more = bc_read_buf(&vm->buffer, vm->exprs.v, &len);
bc_vec_pushByte(&vm->buffer, '\0');
return more;
}
static void
bc_vm_exprs(void)
{
bool clear;
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
clear = true;
vm->mode = BC_MODE_EXPRS;
bc_lex_file(&vm->prs.l, bc_program_exprs_name);
BC_SIG_LOCK;
bc_vec_init(&vm->buffer, sizeof(uchar), BC_DTOR_NONE);
BC_SETJMP_LOCKED(vm, err);
BC_SIG_UNLOCK;
while (bc_vm_readBuf(clear))
{
size_t len = vm->buffer.len - 1;
const char* str = vm->buffer.v;
clear = (len < 2 || str[len - 2] != '\\' || str[len - 1] != '\n');
if (!clear) continue;
bc_vm_process(vm->buffer.v, BC_MODE_EXPRS);
}
if (!clear) bc_vm_process(vm->buffer.v, BC_MODE_EXPRS);
err:
BC_SIG_MAYLOCK;
bc_vm_clean();
if (vm->status == (sig_atomic_t) BC_STATUS_SUCCESS) BC_LONGJMP_STOP;
bc_vec_free(&vm->buffer);
BC_LONGJMP_CONT(vm);
}
#if BC_ENABLED
static void
bc_vm_load(const char* name, const char* text)
{
bc_lex_file(&vm->prs.l, name);
bc_parse_text(&vm->prs, text, BC_MODE_FILE);
BC_SIG_LOCK;
while (vm->prs.l.t != BC_LEX_EOF)
{
vm->parse(&vm->prs);
}
BC_SIG_UNLOCK;
}
#endif
static void
bc_vm_defaultMsgs(void)
{
size_t i;
for (i = 0; i < BC_ERR_IDX_NELEMS + BC_ENABLED; ++i)
{
vm->err_ids[i] = bc_errs[i];
}
for (i = 0; i < BC_ERR_NELEMS; ++i)
{
vm->err_msgs[i] = bc_err_msgs[i];
}
}
static void
bc_vm_gettext(void)
{
#if BC_ENABLE_NLS
uchar id = 0;
int set, msg = 1;
size_t i;
if (vm->locale == NULL)
{
vm->catalog = BC_VM_INVALID_CATALOG;
bc_vm_defaultMsgs();
return;
}
vm->catalog = catopen(BC_MAINEXEC, NL_CAT_LOCALE);
if (vm->catalog == BC_VM_INVALID_CATALOG)
{
bc_vm_defaultMsgs();
return;
}
for (set = 1; msg <= BC_ERR_IDX_NELEMS + BC_ENABLED; ++msg)
{
vm->err_ids[msg - 1] = catgets(vm->catalog, set, msg, bc_errs[msg - 1]);
}
i = 0;
id = bc_err_ids[i];
for (set = id + 2, msg = 1; i < BC_ERR_NELEMS; ++i, ++msg)
{
if (id != bc_err_ids[i])
{
msg = 1;
id = bc_err_ids[i];
set = id + 2;
}
vm->err_msgs[i] = catgets(vm->catalog, set, msg, bc_err_msgs[i]);
}
#else
bc_vm_defaultMsgs();
#endif
}
static void
bc_vm_exec(void)
{
size_t i;
#if DC_ENABLED
bool has_file = false;
#endif
#if BC_ENABLED
if (BC_IS_BC && (vm->flags & BC_FLAG_L))
{
vm->no_redefine = true;
bc_vm_load(bc_lib_name, bc_lib);
#if BC_ENABLE_EXTRA_MATH
if (!BC_IS_POSIX) bc_vm_load(bc_lib2_name, bc_lib2);
#endif
vm->no_redefine = false;
bc_program_exec(&vm->prog);
}
#endif
assert(!BC_ENABLE_OSSFUZZ || BC_EXPR_EXIT == 0);
if (vm->exprs.len)
{
bc_vm_exprs();
if (vm->status != BC_STATUS_SUCCESS ||
(!vm->no_exprs && vm->exit_exprs && BC_EXPR_EXIT))
{
return;
}
}
for (i = 0; i < vm->files.len; ++i)
{
char* path = *((char**) bc_vec_item(&vm->files, i));
if (!strcmp(path, "")) continue;
#if DC_ENABLED
has_file = true;
#endif
bc_vm_file(path);
if (vm->status != BC_STATUS_SUCCESS) return;
}
#if BC_ENABLE_EXTRA_MATH
bc_unveil("/dev/urandom", "r");
bc_unveil("/dev/random", "r");
bc_unveil(NULL, NULL);
#endif
#if BC_ENABLE_HISTORY
if (BC_TTY && !vm->history.badTerm) bc_pledge(bc_pledge_end_history, NULL);
else
#endif
{
bc_pledge(bc_pledge_end, NULL);
}
#if BC_ENABLE_AFL
__AFL_INIT();
#endif
#if BC_ENABLE_OSSFUZZ
if (BC_VM_RUN_STDIN(has_file))
{
bc_vm_load("<stdin>", (const char*) bc_fuzzer_data);
}
#else
if (BC_VM_RUN_STDIN(has_file)) bc_vm_stdin();
#endif
}
BcStatus
bc_vm_boot(int argc, const char* argv[])
{
int ttyin, ttyout, ttyerr;
bool tty;
const char* const env_len = BC_VM_LINE_LENGTH_STR;
const char* const env_args = BC_VM_ENV_ARGS_STR;
const char* const env_exit = BC_VM_EXPR_EXIT_STR;
const char* const env_clamp = BC_VM_DIGIT_CLAMP_STR;
int env_exit_def = BC_VM_EXPR_EXIT_DEF;
int env_clamp_def = BC_VM_DIGIT_CLAMP_DEF;
BcBigDig scale = BC_NUM_BIGDIG_MAX;
BcBigDig env_scale = BC_NUM_BIGDIG_MAX;
BcBigDig ibase = BC_NUM_BIGDIG_MAX;
BcBigDig env_ibase = BC_NUM_BIGDIG_MAX;
BcBigDig obase = BC_NUM_BIGDIG_MAX;
BcBigDig env_obase = BC_NUM_BIGDIG_MAX;
ttyin = isatty(STDIN_FILENO);
ttyout = isatty(STDOUT_FILENO);
ttyerr = isatty(STDERR_FILENO);
tty = (ttyin != 0 && ttyout != 0 && ttyerr != 0);
vm->flags |= ttyin ? BC_FLAG_TTYIN : 0;
vm->flags |= tty ? BC_FLAG_TTY : 0;
vm->flags |= ttyin && ttyout ? BC_FLAG_I : 0;
bc_vm_sigaction();
bc_vm_init();
vm->file = NULL;
bc_vm_gettext();
#if BC_ENABLE_LINE_LIB
bc_file_init(&vm->ferr, stderr, true);
bc_file_init(&vm->fout, stdout, false);
vm->buf = output_bufs;
#else
bc_file_init(&vm->ferr, STDERR_FILENO, output_bufs + BC_VM_STDOUT_BUF_SIZE,
BC_VM_STDERR_BUF_SIZE, true);
bc_file_init(&vm->fout, STDOUT_FILENO, output_bufs, BC_VM_STDOUT_BUF_SIZE,
false);
vm->buf = output_bufs + BC_VM_STDOUT_BUF_SIZE + BC_VM_STDERR_BUF_SIZE;
#endif
vm->line_len = (uint16_t) bc_vm_envLen(env_len);
bc_vm_setenvFlag(env_exit, env_exit_def, BC_FLAG_EXPR_EXIT);
bc_vm_setenvFlag(env_clamp, env_clamp_def, BC_FLAG_DIGIT_CLAMP);
bc_vec_clear(&vm->files);
bc_vec_clear(&vm->exprs);
#if !BC_ENABLE_LIBRARY
bc_slabvec_init(&vm->slabs);
#endif
bc_program_init(&vm->prog);
bc_parse_init(&vm->prs, &vm->prog, BC_PROG_MAIN);
vm->flags |= BC_TTY ? BC_FLAG_P | BC_FLAG_R : 0;
vm->flags |= BC_I ? BC_FLAG_Q : 0;
#if BC_ENABLED
if (BC_IS_BC)
{
char* var = bc_vm_getenv("POSIXLY_CORRECT");
vm->flags |= BC_FLAG_S * (var != NULL);
bc_vm_getenvFree(var);
if (BC_I) bc_vm_setenvFlag("BC_BANNER", BC_DEFAULT_BANNER, BC_FLAG_Q);
}
#endif
if (BC_TTY)
{
const char* const env_tty = BC_VM_TTY_MODE_STR;
int env_tty_def = BC_VM_TTY_MODE_DEF;
const char* const env_prompt = BC_VM_PROMPT_STR;
int env_prompt_def = BC_VM_PROMPT_DEF;
bc_vm_setenvFlag(env_tty, env_tty_def, BC_FLAG_TTY);
bc_vm_setenvFlag(env_prompt, tty ? env_prompt_def : 0, BC_FLAG_P);
#if BC_ENABLE_HISTORY
if (BC_TTY) bc_history_init(&vm->history);
#endif
}
bc_vm_envArgs(env_args, &env_scale, &env_ibase, &env_obase);
bc_args(argc, argv, true, &scale, &ibase, &obase);
{
BC_SIG_UNLOCK;
scale = scale == BC_NUM_BIGDIG_MAX ? env_scale : scale;
#if BC_ENABLED
scale = scale == BC_NUM_BIGDIG_MAX && BC_L ? 20 : scale;
#endif
obase = obase == BC_NUM_BIGDIG_MAX ? env_obase : obase;
ibase = ibase == BC_NUM_BIGDIG_MAX ? env_ibase : ibase;
if (scale != BC_NUM_BIGDIG_MAX)
{
bc_program_assignBuiltin(&vm->prog, true, false, scale);
}
if (obase != BC_NUM_BIGDIG_MAX)
{
bc_program_assignBuiltin(&vm->prog, false, true, obase);
}
if (ibase != BC_NUM_BIGDIG_MAX)
{
bc_program_assignBuiltin(&vm->prog, false, false, ibase);
}
BC_SIG_LOCK;
}
if (BC_I)
{
const char* const env_sigint = BC_VM_SIGINT_RESET_STR;
int env_sigint_def = BC_VM_SIGINT_RESET_DEF;
bc_vm_setenvFlag(env_sigint, env_sigint_def, BC_FLAG_SIGINT);
}
#if BC_ENABLED
if (BC_IS_POSIX) vm->flags &= ~(BC_FLAG_G);
if (BC_IS_BC && BC_I && (vm->flags & BC_FLAG_Q))
{
bc_vm_info(NULL);
bc_file_putchar(&vm->fout, bc_flush_none, '\n');
bc_file_flush(&vm->fout, bc_flush_none);
}
#endif
BC_SIG_UNLOCK;
bc_vm_exec();
BC_SIG_LOCK;
return (BcStatus) vm->status;
}
#endif
void
bc_vm_init(void)
{
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
BC_SIG_ASSERT_LOCKED;
#if !BC_ENABLE_LIBRARY
bc_num_setup(&vm->zero, vm->zero_num, BC_VM_ONE_CAP);
#endif
bc_num_setup(&vm->one, vm->one_num, BC_VM_ONE_CAP);
bc_num_one(&vm->one);
memcpy(vm->max_num, bc_num_bigdigMax,
bc_num_bigdigMax_size * sizeof(BcDig));
memcpy(vm->max2_num, bc_num_bigdigMax2,
bc_num_bigdigMax2_size * sizeof(BcDig));
bc_num_setup(&vm->max, vm->max_num, BC_NUM_BIGDIG_LOG10);
bc_num_setup(&vm->max2, vm->max2_num, BC_NUM_BIGDIG_LOG10);
vm->max.len = bc_num_bigdigMax_size;
vm->max2.len = bc_num_bigdigMax2_size;
vm->maxes[BC_PROG_GLOBALS_IBASE] = BC_NUM_MAX_POSIX_IBASE;
vm->maxes[BC_PROG_GLOBALS_OBASE] = BC_MAX_OBASE;
vm->maxes[BC_PROG_GLOBALS_SCALE] = BC_MAX_SCALE;
#if BC_ENABLE_EXTRA_MATH
vm->maxes[BC_PROG_MAX_RAND] = ((BcRand) 0) - 1;
#endif
#if BC_ENABLED
#if !BC_ENABLE_LIBRARY
if (BC_IS_BC && !BC_IS_POSIX)
#endif
{
vm->maxes[BC_PROG_GLOBALS_IBASE] = BC_NUM_MAX_IBASE;
}
#endif
}
#if BC_ENABLE_LIBRARY
void
bc_vm_atexit(void)
{
#if BC_DEBUG
#if BC_ENABLE_LIBRARY
BcVm* vm = bcl_getspecific();
#endif
#endif
bc_vm_shutdown();
#if BC_DEBUG
bc_vec_free(&vm->jmp_bufs);
#endif
}
#else
BcStatus
bc_vm_atexit(BcStatus status)
{
BcStatus s = BC_STATUS_IS_ERROR(status) ? status : BC_STATUS_SUCCESS;
bc_vm_shutdown();
#if BC_DEBUG
bc_vec_free(&vm->jmp_bufs);
#endif
return s;
}
#endif