cstreamsocket.h

最近在学习directshow, Directshow实务精选的源代码

//
// CStreamSocket.h
//

#ifndef __H_CStreamSocket__
#define __H_CStreamSocket__

class CStreamSocket
{
protected:
	SOCKET		mSocket;
	BOOL		mIsConnected;

	BOOL		mIsReceiving;  // For receiving
	HANDLE		mRcvThread;

	BOOL		mIsSending;    // For sending
	HANDLE		mSndThread;

public:
	CStreamSocket();
	virtual ~CStreamSocket();

public:
	BOOL Attach(SOCKET inSocket);
	void Detach(void);

	BOOL ConnectTo(const char * inTarget, WORD inPort);
	BOOL ConnectTo(long inTarget, WORD inPort);
	BOOL IsConnected(void) { return mIsConnected; };

	// Receive functions
	BOOL StartReceiving(void);
	void StopReceiving(void);
	BOOL IsReceiving(void) { return mIsReceiving; };

	// Send functions
	BOOL StartSending(void);
	void StopSending(void);
	BOOL IsSending(void) { return mIsSending; };

protected:
	static DWORD WINAPI ReceivingThrd(void * pParam);
	static DWORD WINAPI SendingThrd(void * pParam);

	virtual void ReceivingLoop(void);
	virtual void SendingLoop(void);
};

#endif // __H_CStreamSocket__