Ecosystem/p2pool
1
0
Fork 0
mirror of https://github.com/SChernykh/p2pool.git synced 2025-02-02 03:06:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

allow named pipe as stdin

Browse source
This commit is contained in:
T W Lee 2023-03-08 02:35:49 +13:00
parent a76d8049c0
commit c71d89c95a
2 changed files with 32 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
src/console_commands.cpp
View file

@ -36,15 +36,12 @@ ConsoleCommands::ConsoleCommands(p2pool* pool)
, m_loop{} , m_loop{}
, m_shutdownAsync{} , m_shutdownAsync{}
, m_tty{} , m_tty{}
, m_stdin_pipe{}
, m_stdin_handle()
, m_loopThread{} , m_loopThread{}
, m_readBuf{} , m_readBuf{}
, m_readBufInUse(false) , m_readBufInUse(false)
{ {
if (uv_guess_handle(0) != UV_TTY) {
LOGERR(1, "tty is not available");
throw std::exception();
}
int err = uv_loop_init(&m_loop); int err = uv_loop_init(&m_loop);
if (err) { if (err) {
LOGERR(1, "failed to create event loop, error " << uv_err_name(err)); LOGERR(1, "failed to create event loop, error " << uv_err_name(err));
@ -58,14 +55,36 @@ ConsoleCommands::ConsoleCommands(p2pool* pool)
} }
m_shutdownAsync.data = this; m_shutdownAsync.data = this;
err = uv_tty_init(&m_loop, &m_tty, 0, 1); uv_handle_type stdin_type = uv_guess_handle(0);
if (err) { LOGINFO(3, "uv_guess_handle returned " << (int)stdin_type);
LOGERR(1, "uv_tty_init failed, error " << uv_err_name(err)); if (stdin_type == UV_TTY) {
LOGINFO(3, "processing stdin as UV_TTY");
err = uv_tty_init(&m_loop, &m_tty, 0, 1);
if (err) {
LOGERR(1, "uv_tty_init failed, error " << uv_err_name(err));
throw std::exception();
}
m_stdin_handle = reinterpret_cast<uv_handle_t*>(&m_tty);
} else if (stdin_type == UV_NAMED_PIPE) {
LOGINFO(3, "processing stdin as UV_NAMED_PIPE");
err = uv_pipe_init(&m_loop, &m_stdin_pipe, 0);
if (err) {
LOGERR(1, "uv_pipe_init failed, error " << uv_err_name(err));
throw std::exception();
}
m_stdin_handle = reinterpret_cast<uv_handle_t*>(&m_stdin_pipe);
err = uv_pipe_open(&m_stdin_pipe, 0);
if (err) {
LOGERR(1, "uv_pipe_open failed, error " << uv_err_name(err));
throw std::exception();
}
} else {
LOGERR(1, "tty or named pipe is not available");
throw std::exception(); throw std::exception();
} }
m_tty.data = this; m_stdin_handle->data = this;
err = uv_read_start(reinterpret_cast<uv_stream_t*>(&m_tty), allocCallback, stdinReadCallback); err = uv_read_start(reinterpret_cast<uv_stream_t*>(m_stdin_handle), allocCallback, stdinReadCallback);
if (err) { if (err) {
LOGERR(1, "uv_read_start failed, error " << uv_err_name(err)); LOGERR(1, "uv_read_start failed, error " << uv_err_name(err));
throw std::exception(); throw std::exception();

4
src/console_commands.h
View file

@ -35,6 +35,8 @@ private:
uv_loop_t m_loop; uv_loop_t m_loop;
uv_async_t m_shutdownAsync; uv_async_t m_shutdownAsync;
uv_tty_t m_tty; uv_tty_t m_tty;
uv_pipe_t m_stdin_pipe;
uv_handle_t* m_stdin_handle;
uv_thread_t m_loopThread; uv_thread_t m_loopThread;
char m_readBuf[64]; char m_readBuf[64];
@ -48,7 +50,7 @@ private:
{ {
ConsoleCommands* pThis = reinterpret_cast<ConsoleCommands*>(async->data); ConsoleCommands* pThis = reinterpret_cast<ConsoleCommands*>(async->data);
uv_close(reinterpret_cast<uv_handle_t*>(&pThis->m_shutdownAsync), nullptr); uv_close(reinterpret_cast<uv_handle_t*>(&pThis->m_shutdownAsync), nullptr);
uv_close(reinterpret_cast<uv_handle_t*>(&pThis->m_tty), nullptr); uv_close(reinterpret_cast<uv_handle_t*>(pThis->m_stdin_handle), nullptr);
} }
static void allocCallback(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf); static void allocCallback(uv_handle_t* handle, size_t suggested_size, uv_buf_t* buf);