mirror of
https://github.com/sockspls/badfish
synced 2025-05-03 01:59:36 +00:00
Fix Logger under MSVC iostream libraries
We need splitted Tie classes because MSVC stream library takes a lock on buffer both on reading and on writing and this causes an hang because, while searching, the I/O thread is locked on getline() and when main thread is trying to std::cout() something it blocks on the same lock waiting for I/O thread getting some input and releasing the lock. The solution is to use separated streambuf objects for cin and cout. Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
parent
17d1940278
commit
3b7dbc4f6d
1 changed files with 36 additions and 30 deletions
66
src/misc.cpp
66
src/misc.cpp
|
@ -106,40 +106,24 @@ void dbg_print() {
|
||||||
|
|
||||||
|
|
||||||
/// Our fancy logging facility. The trick here is to replace cin.rdbuf() and
|
/// Our fancy logging facility. The trick here is to replace cin.rdbuf() and
|
||||||
/// cout.rdbuf() with this one that tees cin and cout to a file stream. We can
|
/// cout.rdbuf() with two Tie objects that tie cin and cout to a file stream. We
|
||||||
/// toggle the logging of std::cout and std:cin at runtime while preserving i/o
|
/// can toggle the logging of std::cout and std:cin at runtime while preserving
|
||||||
/// functionality and without changing a single line of code!
|
/// usual i/o functionality and without changing a single line of code!
|
||||||
/// Idea from http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81
|
/// Idea from http://groups.google.com/group/comp.lang.c++/msg/1d941c0f26ea0d81
|
||||||
|
|
||||||
class Logger: public streambuf {
|
class Logger {
|
||||||
|
|
||||||
Logger() : cinbuf(cin.rdbuf()), coutbuf(cout.rdbuf()) {}
|
Logger() : in(cin.rdbuf(), file), out(cout.rdbuf(), file) {}
|
||||||
~Logger() { start(false); }
|
~Logger() { start(false); }
|
||||||
|
|
||||||
public:
|
struct Tie: public streambuf { // MSVC requires splitted streambuf for cin and cout
|
||||||
static void start(bool b) {
|
|
||||||
|
|
||||||
static Logger l;
|
Tie(streambuf* b, ofstream& f) : buf(b), file(f) {}
|
||||||
|
|
||||||
if (b && !l.file.is_open())
|
int sync() { return file.rdbuf()->pubsync(), buf->pubsync(); }
|
||||||
{
|
int overflow(int c) { return log(buf->sputc((char)c), "<< "); }
|
||||||
l.file.open("io_log.txt", ifstream::out | ifstream::app);
|
int underflow() { return buf->sgetc(); }
|
||||||
cin.rdbuf(&l);
|
int uflow() { return log(buf->sbumpc(), ">> "); }
|
||||||
cout.rdbuf(&l);
|
|
||||||
}
|
|
||||||
else if (!b && l.file.is_open())
|
|
||||||
{
|
|
||||||
cout.rdbuf(l.coutbuf);
|
|
||||||
cin.rdbuf(l.cinbuf);
|
|
||||||
l.file.close();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
int sync() { return file.rdbuf()->pubsync(), coutbuf->pubsync(); }
|
|
||||||
int overflow(int c) { return log(coutbuf->sputc((char)c), "<< ") ; }
|
|
||||||
int underflow() { return cinbuf->sgetc(); }
|
|
||||||
int uflow() { return log(cinbuf->sbumpc(), ">> "); }
|
|
||||||
|
|
||||||
int log(int c, const char* prefix) {
|
int log(int c, const char* prefix) {
|
||||||
|
|
||||||
|
@ -151,13 +135,35 @@ private:
|
||||||
return last = file.rdbuf()->sputc((char)c);
|
return last = file.rdbuf()->sputc((char)c);
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
streambuf* buf;
|
||||||
|
ofstream& file;
|
||||||
|
};
|
||||||
|
|
||||||
ofstream file;
|
ofstream file;
|
||||||
streambuf *cinbuf, *coutbuf;
|
Tie in, out;
|
||||||
|
|
||||||
|
public:
|
||||||
|
static void start(bool b) {
|
||||||
|
|
||||||
|
static Logger l;
|
||||||
|
|
||||||
|
if (b && !l.file.is_open())
|
||||||
|
{
|
||||||
|
l.file.open("io_log.txt", ifstream::out | ifstream::app);
|
||||||
|
cin.rdbuf(&l.in);
|
||||||
|
cout.rdbuf(&l.out);
|
||||||
|
}
|
||||||
|
else if (!b && l.file.is_open())
|
||||||
|
{
|
||||||
|
cout.rdbuf(l.out.buf);
|
||||||
|
cin.rdbuf(l.in.buf);
|
||||||
|
l.file.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
/// Trampoline helper to avoid moving Logger to misc.h header
|
/// Trampoline helper to avoid moving Logger to misc.h
|
||||||
void start_logger(bool b) { Logger::start(b); }
|
void start_logger(bool b) { Logger::start(b); }
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue