bool BasePacketStreamer::encode(tbnet::Packet* packet, tbnet::DataBuffer* output)
{
bool bret = NULL != packet && NULL != output;
if (bret)
{
tbnet::PacketHeader* header = packet->getPacketHeader();
int32_t old_len = output->getDataLen();
BasePacket* bpacket = dynamic_cast<BasePacket*>(packet);
int32_t iret = TFS_SUCCESS;
int32_t header_length = 0;
int64_t pos = 0;
//v1
if (TFS_PACKET_VERSION_V1 == bpacket->get_version())
{
TfsPacketNewHeaderV1 pheader;
pheader.crc_ = bpacket->get_crc();
pheader.flag_ = TFS_PACKET_FLAG_V1;
pheader.id_ = bpacket->get_id();
pheader.length_ = bpacket->get_data_length();
pheader.type_ = header->_pcode;
pheader.version_ = bpacket->get_version();
header_length = pheader.length();
output->ensureFree(header_length + pheader.length_);
iret = pheader.serialize(output->getFree(), output->getFreeLen(), pos);
}//v2 tbnet
else if (TFS_PACKET_VERSION_V2 == bpacket->get_version())
{
TfsPacketNewHeaderV1 pheader;
pheader.crc_ = bpacket->get_crc();
pheader.flag_ = TFS_PACKET_FLAG_V1;
pheader.id_ = bpacket->getChannelId();
pheader.length_ = bpacket->get_data_length();
pheader.type_ = header->_pcode;
pheader.version_ = bpacket->get_version();
header_length = pheader.length();
output->ensureFree(header_length + pheader.length_);
iret = pheader.serialize(output->getFree(), output->getFreeLen(), pos);
}
else//v0
{
TfsPacketNewHeaderV1 pheader;
memset(&pheader, 0, sizeof(pheader));
pheader.flag_ = TFS_PACKET_FLAG_V0;
char* header_data = reinterpret_cast<char*>(&pheader);
int32_t length = TFS_PACKET_HEADER_V0_SIZE - 2;
for (int32_t i = 0; i < length; i++)
{
pheader.version_ = static_cast<uint16_t>(*(header_data + i));
}
header_length = pheader.length();
output->ensureFree(header_length + pheader.length_);
iret = pheader.serialize(output->getFree(), output->getFreeLen(), pos);
}
bret = TFS_SUCCESS == iret;
if (bret)
{
//TBSYS_LOG(DEBUG, "pcode: %d, header length: %d, body length : %" PRI64_PREFIX "d", bpacket->getPCode(), header_length, bpacket->get_data_length());
//Func::hex_dump(output->getData(), output->getDataLen());
output->pourData(header_length);
bret = bpacket->encode(output);
//Func::hex_dump(output->getData(), output->getDataLen());
}
else
{
TBSYS_LOG(ERROR, "encode erorr, pcode: %d", header->_pcode);
output->stripData(output->getDataLen() - old_len);
}
}
return bret;
}
