From c8479ceff6aea37def538c902324c68930947f21 Mon Sep 17 00:00:00 2001 From: _ov4 Date: Sun, 18 Oct 2026 09:42:45 +0000 Subject: add self pipe, and non-block-input --- src/util/non_block_io.cc | 49 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 src/util/non_block_io.cc (limited to 'src/util') diff --git a/src/util/non_block_io.cc b/src/util/non_block_io.cc new file mode 100644 index 0000000..31edbc9 --- /dev/null +++ b/src/util/non_block_io.cc @@ -0,0 +1,49 @@ +#include "util/non_block_input.h" + +#include +#include + +#include "pipe.h" + +using namespace std; + +namespace ov4 +{ + + +/* + * read from both stdin, and self pipe + * returns immediately as soon as stdin or pipe is "readable" + * returns read()'s ret value + * and set src to the the read source (stdin or pipe) + */ +int non_block_read_char(char *c, int *src) +{ + // get fd check range + int max_fd; + if (STDIN_FILENO > self_pipe_fd[0]) max_fd = STDIN_FILENO; + else max_fd = self_pipe_fd[0]; + + // monitor STDIN, and self pipe + fd_set rset; + FD_ZERO(&rset); + FD_SET(STDIN_FILENO, &rset); + FD_SET(self_pipe_fd[0], &rset); + + select(max_fd+1, &rset, nullptr, nullptr, nullptr); + + int ret = -1; + if (FD_ISSET(self_pipe_fd[0], &rset)) + { + ret = read(self_pipe_fd[0], c, 1); + if (src != nullptr) *src = self_pipe_fd[0]; + } + else if (FD_ISSET(STDIN_FILENO, &rset)) + { + ret = read(STDIN_FILENO, c, 1); + if (src != nullptr) *src = self_pipe_fd[0]; + } + return ret; +} + +} \ No newline at end of file -- cgit v1.3.1-4-g156e