mirror of
https://github.com/bitcoin/bitcoin.git
synced 2024-11-20 02:25:40 +01:00
add stdin helpers for password input support
This commit is contained in:
parent
b0e268d2f6
commit
0da503e947
@ -520,6 +520,8 @@ endif
|
||||
libbitcoin_cli_a_CPPFLAGS = $(AM_CPPFLAGS) $(BITCOIN_INCLUDES)
|
||||
libbitcoin_cli_a_CXXFLAGS = $(AM_CXXFLAGS) $(PIE_FLAGS)
|
||||
libbitcoin_cli_a_SOURCES = \
|
||||
compat/stdin.h \
|
||||
compat/stdin.cpp \
|
||||
rpc/client.cpp \
|
||||
$(BITCOIN_CORE_H)
|
||||
|
||||
|
72
src/compat/stdin.cpp
Normal file
72
src/compat/stdin.cpp
Normal file
@ -0,0 +1,72 @@
|
||||
// Copyright (c) 2018 The Bitcoin Core developers
|
||||
// Distributed under the MIT software license, see the accompanying
|
||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
#if defined(HAVE_CONFIG_H)
|
||||
#include <config/bitcoin-config.h>
|
||||
#endif
|
||||
|
||||
#include <cstdio> // for fileno(), stdin
|
||||
|
||||
#ifdef WIN32
|
||||
#include <windows.h> // for SetStdinEcho()
|
||||
#include <io.h> // for isatty()
|
||||
#else
|
||||
#include <termios.h> // for SetStdinEcho()
|
||||
#include <unistd.h> // for SetStdinEcho(), isatty()
|
||||
#include <sys/poll.h> // for StdinReady()
|
||||
#endif
|
||||
|
||||
#include <compat/stdin.h>
|
||||
|
||||
// https://stackoverflow.com/questions/1413445/reading-a-password-from-stdcin
|
||||
void SetStdinEcho(bool enable)
|
||||
{
|
||||
#ifdef WIN32
|
||||
HANDLE hStdin = GetStdHandle(STD_INPUT_HANDLE);
|
||||
DWORD mode;
|
||||
GetConsoleMode(hStdin, &mode);
|
||||
if (!enable) {
|
||||
mode &= ~ENABLE_ECHO_INPUT;
|
||||
} else {
|
||||
mode |= ENABLE_ECHO_INPUT;
|
||||
}
|
||||
SetConsoleMode(hStdin, mode);
|
||||
#else
|
||||
struct termios tty;
|
||||
tcgetattr(STDIN_FILENO, &tty);
|
||||
if (!enable) {
|
||||
tty.c_lflag &= ~ECHO;
|
||||
} else {
|
||||
tty.c_lflag |= ECHO;
|
||||
}
|
||||
(void)tcsetattr(STDIN_FILENO, TCSANOW, &tty);
|
||||
#endif
|
||||
}
|
||||
|
||||
bool StdinTerminal()
|
||||
{
|
||||
#ifdef WIN32
|
||||
return _isatty(_fileno(stdin));
|
||||
#else
|
||||
return isatty(fileno(stdin));
|
||||
#endif
|
||||
}
|
||||
|
||||
bool StdinReady()
|
||||
{
|
||||
if (!StdinTerminal()) {
|
||||
return true;
|
||||
}
|
||||
#ifdef WIN32
|
||||
return false;
|
||||
#else
|
||||
struct pollfd fds;
|
||||
fds.fd = 0; /* this is STDIN */
|
||||
fds.events = POLLIN;
|
||||
return poll(&fds, 1, 0) == 1;
|
||||
#endif
|
||||
}
|
||||
|
||||
NoechoInst::NoechoInst() { SetStdinEcho(false); }
|
||||
NoechoInst::~NoechoInst() { SetStdinEcho(true); }
|
18
src/compat/stdin.h
Normal file
18
src/compat/stdin.h
Normal file
@ -0,0 +1,18 @@
|
||||
// Copyright (c) 2018 The Bitcoin Core developers
|
||||
// Distributed under the MIT software license, see the accompanying
|
||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||
|
||||
#ifndef BITCOIN_COMPAT_STDIN_H
|
||||
#define BITCOIN_COMPAT_STDIN_H
|
||||
|
||||
struct NoechoInst {
|
||||
NoechoInst();
|
||||
~NoechoInst();
|
||||
};
|
||||
|
||||
#define NO_STDIN_ECHO() NoechoInst _no_echo
|
||||
|
||||
bool StdinTerminal();
|
||||
bool StdinReady();
|
||||
|
||||
#endif // BITCOIN_COMPAT_STDIN_H
|
Loading…
Reference in New Issue
Block a user