hardcoresushi/libcryfs
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Make DontEchoStdinToStdoutRAII work on windows

Browse Source
This commit is contained in:
Sebastian Messmer 2018-05-19 06:29:28 -07:00
parent 22f729bcab
commit e5d8bf82c3
1 changed files with 49 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/cpp-utils/io/DontEchoStdinToStdoutRAII.h
View File

@ -2,22 +2,25 @@
#ifndef MESSMER_CPPUTILS_IO_DONTECHOSTDINTOSTDOUTRAII_H
#define MESSMER_CPPUTILS_IO_DONTECHOSTDINTOSTDOUTRAII_H
#include <iostream>
#include <string>
#include <termios.h>
#include <unistd.h>
#include <memory>
#include "../macros.h"
namespace cpputils {
/**
* If you create an instance of this class in your scope, then any user input from stdin
* won't be echoed back to stdout until the instance leaves the scope.
* This can be very handy for password inputs where you don't want the password to be visible on screen.
*/
#if !defined(_MSC_VER)
#include <termios.h>
#include <unistd.h>
namespace cpputils {
class DontEchoStdinToStdoutRAII final {
public:
DontEchoStdinToStdoutRAII(): _old_state() {
DontEchoStdinToStdoutRAII() : _old_state() {
tcgetattr(STDIN_FILENO, &_old_state);
termios new_state = _old_state;
new_state.c_lflag &= ~ECHO;
@ -36,4 +39,33 @@ private:
}
#else
#include <windows.h>
namespace cpputils {
class DontEchoStdinToStdoutRAII final {
public:
DontEchoStdinToStdoutRAII() : _old_state() {
HANDLE hStdin = GetStdHandle(STD_INPUT_HANDLE);
GetConsoleMode(hStdin, &_old_state);
SetConsoleMode(hStdin, _old_state & (~ENABLE_ECHO_INPUT));
}
~DontEchoStdinToStdoutRAII() {
HANDLE hStdin = GetStdHandle(STD_INPUT_HANDLE);
SetConsoleMode(hStdin, _old_state);
}
private:
DWORD _old_state;
DISALLOW_COPY_AND_ASSIGN(DontEchoStdinToStdoutRAII);
};
}
#endif
#endif