1 files changed, 71 insertions, 0 deletions
diff --git a/lib/common/SelfFlushingStream.h b/lib/common/SelfFlushingStream.h
new file mode 100644
index 00000000..36e9a4d3
--- /dev/null
+++ b/lib/common/SelfFlushingStream.h
@@ -0,0 +1,71 @@
+// --------------------------------------------------------------------------
+//
+// File
+//		Name:    SelfFlushingStream.h
+//		Purpose: A stream wrapper that always flushes the underlying
+//			 stream, to ensure protocol safety.
+//		Created: 2008/08/20
+//
+// --------------------------------------------------------------------------
+
+#ifndef SELFFLUSHINGSTREAM__H
+#define SELFFLUSHINGSTREAM__H
+
+#include "IOStream.h"
+
+// --------------------------------------------------------------------------
+//
+// Class
+//		Name:    SelfFlushingStream
+//		Purpose: A stream wrapper that always flushes the underlying
+//			 stream, to ensure protocol safety.
+//		Created: 2008/08/20
+//
+// --------------------------------------------------------------------------
+class SelfFlushingStream : public IOStream
+{
+public:
+	SelfFlushingStream(IOStream &rSource)
+	: mrSource(rSource) { }
+
+	SelfFlushingStream(const SelfFlushingStream &rToCopy)
+	: mrSource(rToCopy.mrSource) { }
+	
+	~SelfFlushingStream()
+	{
+		Flush();
+	}
+
+private:
+	// no copying from IOStream allowed
+	SelfFlushingStream(const IOStream& rToCopy);
+	
+public:
+	virtual int Read(void *pBuffer, int NBytes,
+		int Timeout = IOStream::TimeOutInfinite)
+	{
+		return mrSource.Read(pBuffer, NBytes, Timeout);
+	}
+	virtual pos_type BytesLeftToRead()
+	{
+		return mrSource.BytesLeftToRead();
+	}
+	virtual void Write(const void *pBuffer, int NBytes)
+	{
+		mrSource.Write(pBuffer, NBytes);
+	}
+	virtual bool StreamDataLeft()
+	{
+		return mrSource.StreamDataLeft();
+	}
+	virtual bool StreamClosed()
+	{
+		return mrSource.StreamClosed();
+	}
+
+private:
+	IOStream &mrSource;
+};
+
+#endif // SELFFLUSHINGSTREAM__H
+