#ifndef WIN32
#include <errno.h>
+#include <unistd.h>
#include <fcntl.h>
#include <termios.h>
#include <sys/ioctl.h>
namespace {
#ifdef WIN32
-DWORD stream_to_sys(Console::Stream stream)
+DWORD stream_to_sys(Msp::IO::Console::Stream stream)
{
switch(stream)
{
- case INPUT: return STD_INPUT_HANDLE;
- case OUTPUT: return STD_OUTPUT_HANDLE;
- case ERROR: return STD_ERROR_HANDLE;
+ case Msp::IO::Console::CIN: return STD_INPUT_HANDLE;
+ case Msp::IO::Console::COUT: return STD_OUTPUT_HANDLE;
+ case Msp::IO::Console::CERR: return STD_ERROR_HANDLE;
default: throw invalid_argument("stream_to_sys");
}
}
Console::Console(Stream s):
stream(s)
{
- mode = (stream==INPUT ? M_READ : M_WRITE);
+ mode = (stream==CIN ? M_READ : M_WRITE);
#ifdef WIN32
*handle = GetStdHandle(stream_to_sys(stream));
#else
*handle = stream;
- if(stream==INPUT)
+ if(stream==CIN)
tcgetattr(*handle, &orig_attr);
#endif
- if(stream==INPUT)
+ if(stream==CIN)
set_events(P_INPUT);
}
Console::~Console()
{
#ifndef WIN32
- if(stream==INPUT)
+ if(stream==CIN)
tcsetattr(*handle, TCSADRAIN, &orig_attr);
#endif
}
#ifdef WIN32
DWORD m;
- GetConsoleMode(*handle, &m);
- SetConsoleMode(*handle, (m&~ENABLE_LINE_INPUT) | (l?ENABLE_LINE_INPUT:0));
+ if(!GetConsoleMode(*handle, &m))
+ throw system_error("GetConsoleMode");
+ if(!SetConsoleMode(*handle, (m&~ENABLE_LINE_INPUT) | (l?ENABLE_LINE_INPUT:0)))
+ throw system_error("SetConsoleMode");
#else
- // XXX ICANON does more than just set line buffering, may need a bit more thought
termios t;
- tcgetattr(*handle, &t);
+ if(tcgetattr(*handle, &t)==-1)
+ throw system_error("tcgetattr");
t.c_lflag = (t.c_lflag&~ICANON) | (l?ICANON:0);
- tcsetattr(*handle, TCSADRAIN, &t);
+ // man termios warns that VMIN and VTIME may share indices with VEOF and VEOL
+ if(l)
+ {
+ t.c_cc[VEOF] = orig_attr.c_cc[VEOF];
+ t.c_cc[VEOL] = orig_attr.c_cc[VEOL];
+ }
+ else
+ {
+ t.c_cc[VMIN] = 1;
+ t.c_cc[VTIME] = 0;
+ }
+ if(tcsetattr(*handle, TCSADRAIN, &t)==-1)
+ throw system_error("tcsetattr");
#endif
}
check_access(M_WRITE);
#ifdef WIN32
- // XXX Figure out how to do this
- rows = 24;
- cols = 80;
+ CONSOLE_SCREEN_BUFFER_INFO sbi;
+ if(!GetConsoleScreenBufferInfo(*handle, &sbi))
+ throw system_error("GetConsoleScreenBufferInfo");
+ // Right/bottom coords are inclusive
+ rows = sbi.srWindow.Bottom+1-sbi.srWindow.Top;
+ cols = sbi.srWindow.Right+1-sbi.srWindow.Left;
#else
struct winsize wsz;
- ioctl(*handle, TIOCGWINSZ, &wsz);
+ if(ioctl(*handle, TIOCGWINSZ, &wsz)==-1)
+ throw system_error("ioctl TIOCGWINSZ");
rows = wsz.ws_row;
cols = wsz.ws_col;
#endif
}
+void Console::redirect(Base &other)
+{
+ Handle other_handle = other.get_handle(mode&M_RDWR);
+#ifdef WIN32
+ SetStdHandle(stream_to_sys(stream), *other_handle);
+#else
+ dup2(*other_handle, *handle);
+#endif
+}
+
unsigned Console::do_write(const char *buf, unsigned len)
{
check_access(M_WRITE);
Console &Console::instance(Stream s)
{
- static Console in(INPUT);
- static Console out(OUTPUT);
- static Console err(ERROR);
+ static Console in(CIN);
+ static Console out(COUT);
+ static Console err(CERR);
switch(s)
{
- case INPUT: return in;
- case OUTPUT: return out;
- case ERROR: return err;
+ case CIN: return in;
+ case COUT: return out;
+ case CERR: return err;
}
throw invalid_argument("Console::instance");
}
-Console &cin = Console::instance(Console::INPUT);
-Console &cout = Console::instance(Console::OUTPUT);
-Console &cerr = Console::instance(Console::ERROR);
+Console &cin = Console::instance(Console::CIN);
+Console &cout = Console::instance(Console::COUT);
+Console &cerr = Console::instance(Console::CERR);
} // namespace IO
} // namespace Msp