/* */ #include "BtHaveAllMessage.h" #include "DlAbortEx.h" #include "PeerMessageUtil.h" #include "message.h" #include "Peer.h" namespace aria2 { BtHaveAllMessageHandle BtHaveAllMessage::create(const unsigned char* data, size_t dataLength) { if(dataLength != 1) { throw new DlAbortEx(EX_INVALID_PAYLOAD_SIZE, "have all", dataLength, 1); } uint8_t id = PeerMessageUtil::getId(data); if(id != ID) { throw new DlAbortEx(EX_INVALID_BT_MESSAGE_ID, id, "have all", ID); } BtHaveAllMessageHandle message = new BtHaveAllMessage(); return message; } void BtHaveAllMessage::doReceivedAction() { if(!peer->isFastExtensionEnabled()) { throw new DlAbortEx("%s received while fast extension is disabled", toString().c_str()); } peer->setAllBitfield(); } size_t BtHaveAllMessage::MESSAGE_LENGTH = 5; const unsigned char* BtHaveAllMessage::getMessage() { if(!msg) { /** * len --- 1, 4bytes * id --- 14, 1byte * total: 5bytes */ msg = new unsigned char[MESSAGE_LENGTH]; PeerMessageUtil::createPeerMessageString(msg, MESSAGE_LENGTH, 1, ID); } return msg; } size_t BtHaveAllMessage::getMessageLength() { return MESSAGE_LENGTH; } std::string BtHaveAllMessage::toString() const { return "have all"; } } // namespace aria2