diff options
author | Stefan Eßer <se@FreeBSD.org> | 2020-06-27 15:03:19 +0000 |
---|---|---|
committer | Stefan Eßer <se@FreeBSD.org> | 2020-06-27 15:03:19 +0000 |
commit | 1f958cfad78842ab9a1193471589231e25596cb3 (patch) | |
tree | 4bbff8044605fcfff11c9d322bb6f53495e4faa7 /src/read.c |
Diffstat (limited to 'src/read.c')
-rw-r--r-- | src/read.c | 226 |
1 files changed, 226 insertions, 0 deletions
diff --git a/src/read.c b/src/read.c new file mode 100644 index 000000000000..c63952523181 --- /dev/null +++ b/src/read.c @@ -0,0 +1,226 @@ +/* + * ***************************************************************************** + * + * Copyright (c) 2018-2020 Gavin D. Howard and contributors. + * + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * * Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + * + * ***************************************************************************** + * + * Code to handle special I/O for bc. + * + */ + +#include <assert.h> +#include <ctype.h> +#include <errno.h> +#include <stdlib.h> +#include <string.h> + +#include <signal.h> + +#include <fcntl.h> +#include <sys/stat.h> +#include <unistd.h> + +#include <read.h> +#include <history.h> +#include <program.h> +#include <vm.h> + +static bool bc_read_binary(const char *buf, size_t size) { + + size_t i; + + for (i = 0; i < size; ++i) { + if (BC_ERR(BC_READ_BIN_CHAR(buf[i]))) return true; + } + + return false; +} + +static bool bc_read_buf(BcVec *vec) { + + char *nl; + + if (!vm.buf_len) return false; + + nl = strchr(vm.buf, '\n'); + + if (nl != NULL) { + + size_t nllen = (size_t) ((nl + 1) - vm.buf); + + nllen = vm.buf_len >= nllen ? nllen : vm.buf_len; + + bc_vec_npush(vec, nllen, vm.buf); + vm.buf_len -= nllen; + memmove(vm.buf, nl + 1, vm.buf_len); + + return true; + } + + bc_vec_npush(vec, vm.buf_len, vm.buf); + vm.buf_len = 0; + + return false; +} + +BcStatus bc_read_chars(BcVec *vec, const char *prompt) { + + bool done = false; + + assert(vec != NULL && vec->size == sizeof(char)); + + BC_SIG_ASSERT_NOT_LOCKED; + + bc_vec_npop(vec, vec->len); + +#if BC_ENABLE_PROMPT + if (BC_USE_PROMPT) { + bc_file_puts(&vm.fout, prompt); + bc_file_flush(&vm.fout); + } +#endif // BC_ENABLE_PROMPT + + if (bc_read_buf(vec)) { + bc_vec_pushByte(vec, '\0'); + return BC_STATUS_SUCCESS; + } + + while (!done) { + + ssize_t r; + + BC_SIG_LOCK; + + r = read(STDIN_FILENO, vm.buf + vm.buf_len, + BC_VM_STDIN_BUF_SIZE - vm.buf_len); + + if (BC_UNLIKELY(r < 0)) { + + if (errno == EINTR) { + + if (vm.status == (sig_atomic_t) BC_STATUS_QUIT) { + BC_SIG_UNLOCK; + return BC_STATUS_QUIT; + } + + assert(vm.sig); + + vm.status = (sig_atomic_t) BC_STATUS_SUCCESS; +#if BC_ENABLE_PROMPT + if (BC_USE_PROMPT) bc_file_puts(&vm.fout, prompt); +#endif // BC_ENABLE_PROMPT + bc_file_flush(&vm.fout); + + BC_SIG_UNLOCK; + + continue; + } + + BC_SIG_UNLOCK; + + bc_vm_err(BC_ERROR_FATAL_IO_ERR); + } + + BC_SIG_UNLOCK; + + if (r == 0) { + bc_vec_pushByte(vec, '\0'); + return BC_STATUS_EOF; + } + + vm.buf_len += (size_t) r; + + done = bc_read_buf(vec); + } + + bc_vec_pushByte(vec, '\0'); + + return BC_STATUS_SUCCESS; +} + +BcStatus bc_read_line(BcVec *vec, const char *prompt) { + + BcStatus s; + +#if BC_ENABLE_HISTORY + if (BC_TTY && !vm.history.badTerm) + s = bc_history_line(&vm.history, vec, prompt); + else s = bc_read_chars(vec, prompt); +#else // BC_ENABLE_HISTORY + s = bc_read_chars(vec, prompt); +#endif // BC_ENABLE_HISTORY + + if (BC_ERR(bc_read_binary(vec->v, vec->len - 1))) + bc_vm_verr(BC_ERROR_FATAL_BIN_FILE, bc_program_stdin_name); + + return s; +} + +void bc_read_file(const char *path, char **buf) { + + BcError e = BC_ERROR_FATAL_IO_ERR; + size_t size, r; + struct stat pstat; + int fd; + + BC_SIG_ASSERT_LOCKED; + + assert(path != NULL); + + fd = open(path, O_RDONLY); + if (BC_ERR(fd < 0)) bc_vm_verr(BC_ERROR_FATAL_FILE_ERR, path); + if (BC_ERR(fstat(fd, &pstat) == -1)) goto malloc_err; + + if (BC_ERR(S_ISDIR(pstat.st_mode))) { + e = BC_ERROR_FATAL_PATH_DIR; + goto malloc_err; + } + + size = (size_t) pstat.st_size; + *buf = bc_vm_malloc(size + 1); + + r = (size_t) read(fd, *buf, size); + if (BC_ERR(r != size)) goto read_err; + + (*buf)[size] = '\0'; + + if (BC_ERR(bc_read_binary(*buf, size))) { + e = BC_ERROR_FATAL_BIN_FILE; + goto read_err; + } + + close(fd); + + return; + +read_err: + free(*buf); +malloc_err: + close(fd); + bc_vm_verr(e, path); +} |