barrier/lib/synergy/CInputPacketStream.h

#ifndef CINPUTPACKETSTREAM_H
#define CINPUTPACKETSTREAM_H

#include "CInputStreamFilter.h"
#include "CBufferedInputStream.h"
#include "CMutex.h"

//! Packetizing input stream filter
/*!
Filters an input stream to extract packet by packet.
*/
class CInputPacketStream : public CInputStreamFilter {
public:
	CInputPacketStream(IInputStream*, bool adoptStream = true);
	~CInputPacketStream();

	// IInputStream overrides
	virtual void		close();
	virtual UInt32		read(void*, UInt32 maxCount, double timeout);
	virtual UInt32		getSize() const;

private:
	enum EResult { kData, kHungup, kTimedout };

	UInt32				getSizeNoLock() const;
	EResult				waitForFullMessage(double timeout) const;
	EResult				getMoreMessage(double timeout) const;
	bool				hasFullMessage() const;

private:
	CMutex				m_mutex;
	mutable UInt32					m_size;
	mutable CBufferedInputStream	m_buffer;
};

#endif