csocketreceiver.cpp

来自「最近在学习directshow, Directshow实务精选的源代码」· C++ 代码 · 共 60 行

//
// CSocketReceiver.cpp
//

/*-----------------------------------------------------*\
			HQ Tech, Make Technology Easy!       
 More information, please go to http://hqtech.nease.net.
/*-----------------------------------------------------*/

#include "stdafx.h"
#include "CSocketReceiver.h"

#include "Netdefs.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

//////////////////////////////////////////////////////////////////////////////
CSocketReceiver::CSocketReceiver()
{
}

CSocketReceiver::~CSocketReceiver()
{
}

void CSocketReceiver::ReceivingLoop(void)
{
	char   buf[1024];
	int    bytes = 0;
	Net_Header * pHeader = (Net_Header *) buf;

	while (mIsReceiving)
	{
		bytes = recv(mSocket, buf, sizeof(Net_Header), 0);
		if (bytes == SOCKET_ERROR || bytes == 0)
		{
			Detach();
			mIsReceiving = FALSE;
			break;
		}

		pHeader->my_ntoh();
		bytes = recv(mSocket, buf, pHeader->pack_size, 0);
		if (bytes == SOCKET_ERROR || bytes == 0)
		{
			Detach();
			mIsReceiving = FALSE;
			break;
		}
		
		buf[bytes]  = '\0';
		// Pop up a message box
		CString msg = "Received content:\n";
		AfxMessageBox(msg + buf);
	}
}