C++ (Cpp) ByteBuffer::Alloc示例

示例#1

文件： UdpComm.cpp 项目： prsood/trochilus

BOOL UdpComm::SendAndRecv( ULONG targetIP, const LPBYTE pSendData, DWORD dwSendSize, LPBYTE* pRecvData, DWORD& dwRecvSize )
{
	UDP_HEADER sendHead;
	sendHead.flag = UDP_FLAG;
	sendHead.nSize = dwSendSize;

	BOOL ret = FALSE;

	do 
	{
		if (! Send( targetIP, (PBYTE)&sendHead, sizeof(UDP_HEADER))) break;

		if (m_isSecure)
			XFC(pSendData,dwSendSize,pSendData,m_xorKey1,m_xorKey2);

		if (! Send( targetIP, pSendData, dwSendSize)) break;

		UDP_HEADER recvHead = {0};

		if ( !ReceiveAll(m_sock,(char*)&recvHead, sizeof(UDP_HEADER)))
			break;


		ByteBuffer buffer;
		buffer.Alloc(recvHead.nSize);

		if (! ReceiveAll(m_sock,(LPBYTE)buffer,recvHead.nSize))
			break;

		//¸´ÖÆÊý¾Ý
		*pRecvData = Alloc(recvHead.nSize);
		memcpy(*pRecvData, (LPBYTE)buffer, recvHead.nSize);
		dwRecvSize =  recvHead.nSize;

		if(m_isSecure)
			XFC(*pRecvData,recvHead.nSize,*pRecvData,m_xorKey1,m_xorKey2);

		ret = TRUE;

	} while (FALSE);

	if ( !ret )
		m_isConnected = FALSE;

	return ret;
}

示例#2

文件： UdpComm.cpp 项目： prsood/trochilus

BOOL UdpComm::Send( ULONG targetIP, const LPBYTE pData, DWORD dwSize )
{
	IN_ADDR addr;
	addr.S_un.S_addr = targetIP;

	ByteBuffer sendByteBuffer;
	sendByteBuffer.Alloc(dwSize);
	memcpy((LPBYTE)sendByteBuffer, pData, dwSize);

	if ( !m_isConnected )
		m_isConnected = Connect(targetIP, g_ConfigInfo.nPort);

	if ( m_isConnected )
		m_isConnected = SendAll(m_sock,(LPBYTE)sendByteBuffer,sendByteBuffer.Size());

	return m_isConnected;
}

示例#3

文件： MyFile.cpp 项目： c4bbage/trochilus

DWORD MyFile::Read( DWORD dwToRead, ByteBuffer& buffer )
{
	if (0 == dwToRead) return 0;

	buffer.Alloc(dwToRead);
	DWORD dwRead = 0;
	BOOL bRet = ::ReadFile(m_handle, (LPBYTE)buffer, dwToRead, &dwRead, NULL);
	if (bRet)
	{
		return dwRead;
	}
	else
	{
		buffer.Free();
		return 0;
	}
}

示例#4

文件： TcpComm.cpp 项目： prsood/trochilus

BOOL TcpComm::Send( ULONG targetIP, const LPBYTE pData, DWORD dwSize )
{
	IN_ADDR addr;
	addr.S_un.S_addr = targetIP;


	ByteBuffer sendByteBuffer;
	sendByteBuffer.Alloc(dwSize);
	memcpy((LPBYTE)sendByteBuffer, pData, dwSize);

	BOOL bSentOK = FALSE;

	if( !m_isConnected)
		m_isConnected = Connect(targetIP);
	
	if (m_isConnected)
		m_isConnected = m_sock.SendAll((LPBYTE)sendByteBuffer, sendByteBuffer.Size());

	return m_isConnected;
}

示例#5

文件： CommManager.cpp 项目： redblack168/trochilus

BOOL CommManager::HandleMessageAndReply( SOCKADDR_IN fromAddr, const LPBYTE pData, DWORD dwDataSize, COMM_NAME commName, BOOL& bValidData, DWORD replyMaxDataSize, ByteBuffer& replyBuffer )
{
	CPGUID cpguid;
	bValidData = HandleMessage(fromAddr, pData, dwDataSize, commName, cpguid);
	if (! bValidData)
	{
		return FALSE;
	}

	//找到需要发送的消息
	ByteBuffer toSendData;
	if (! m_cp.GetMessageToSendById(cpguid, replyMaxDataSize, toSendData))
	{
		m_cp.CreateEmptyPacket(toSendData);
	}
	
	replyBuffer.Alloc(toSendData.Size());
	BOOL bEncryptOK = XorFibonacciCrypt((LPBYTE)toSendData, toSendData.Size(), (LPBYTE)replyBuffer, 2, 7);

	return bEncryptOK;
}

示例#6

文件： CutupProtocol.cpp 项目： c4bbage/trochilus

void CutupProtocol::MergePackets( const PacketList& packetList, ByteBuffer& byteData ) const
{
	DWORD dwDataSize = 0;
	PacketList::const_iterator iter = packetList.begin();
	for (; iter != packetList.end(); iter++)
	{
		dwDataSize += (*iter)->header.size;
	}
	
	byteData.Alloc(dwDataSize);
	LPBYTE pData = byteData;

	iter = packetList.begin();
	for (; iter != packetList.end(); iter++)
	{
		if ((*iter)->header.size > 0)
		{
			memcpy(pData, (*iter)->data, (*iter)->header.size);
			pData += (*iter)->header.size;
		}
	}
}

示例#7

文件： TcpComm.cpp 项目： c4bbage/trochilus

BOOL TcpComm::SendAndRecv( ULONG targetIP, const LPBYTE pSendData, DWORD dwSendSize, LPBYTE* pRecvData, DWORD& dwRecvSize )
{
	TCP_HEADER sendHead;
	sendHead.flag = TCP_FLAG;
	sendHead.nSize = dwSendSize;

	if (! Send(m_sock, targetIP, (PBYTE)&sendHead, sizeof(TCP_HEADER))) return FALSE;

	if (! Send(m_sock, targetIP, pSendData, dwSendSize)) return FALSE;

	TCP_HEADER recvHead = {0};

	int iRecv = m_sock.ReceiveAll((LPBYTE)&recvHead, sizeof(TCP_HEADER));
	
	if (iRecv < 0)
	{
		errorLog(_T("recv http failed WE%d"), ::WSAGetLastError());
	}


	ByteBuffer buffer;
	buffer.Alloc(recvHead.nSize);

	iRecv = m_sock.ReceiveAll(buffer,recvHead.nSize);

	if (iRecv < 0)
	{
		errorLog(_T("recv http failed WE%d"), ::WSAGetLastError());
	}

	//¸´ÖÆÊý¾Ý
	*pRecvData = Alloc(recvHead.nSize);
	memcpy(*pRecvData, (LPBYTE)buffer, recvHead.nSize);
	dwRecvSize =  recvHead.nSize;
	
	return TRUE;
}