diff options
Diffstat (limited to 'lib/base/stdiostream.cpp')
-rw-r--r-- | lib/base/stdiostream.cpp | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/lib/base/stdiostream.cpp b/lib/base/stdiostream.cpp new file mode 100644 index 0000000..449036f --- /dev/null +++ b/lib/base/stdiostream.cpp @@ -0,0 +1,57 @@ +/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */ + +#include "base/stdiostream.hpp" +#include "base/objectlock.hpp" + +using namespace icinga; + +/** + * Constructor for the StdioStream class. + * + * @param innerStream The inner stream. + * @param ownsStream Whether the new object owns the inner stream. If true + * the stream's destructor deletes the inner stream. + */ +StdioStream::StdioStream(std::iostream *innerStream, bool ownsStream) + : m_InnerStream(innerStream), m_OwnsStream(ownsStream) +{ } + +StdioStream::~StdioStream() +{ + Close(); +} + +size_t StdioStream::Read(void *buffer, size_t size, bool allow_partial) +{ + ObjectLock olock(this); + + m_InnerStream->read(static_cast<char *>(buffer), size); + return m_InnerStream->gcount(); +} + +void StdioStream::Write(const void *buffer, size_t size) +{ + ObjectLock olock(this); + + m_InnerStream->write(static_cast<const char *>(buffer), size); +} + +void StdioStream::Close() +{ + Stream::Close(); + + if (m_OwnsStream) { + delete m_InnerStream; + m_OwnsStream = false; + } +} + +bool StdioStream::IsDataAvailable() const +{ + return !IsEof(); +} + +bool StdioStream::IsEof() const +{ + return !m_InnerStream->good(); +} |