/* */ #ifndef _D_BT_PIECE_MESSAGE_VALIDATOR_H_ #define _D_BT_PIECE_MESSAGE_VALIDATOR_H_ #include "BtMessageValidator.h" #include "BtPieceMessage.h" #include "PeerMessageUtil.h" namespace aria2 { class BtPieceMessageValidator : public BtMessageValidator { private: const BtPieceMessage* message; size_t numPiece; size_t pieceLength; public: BtPieceMessageValidator(const BtPieceMessage* message, size_t numPiece, size_t pieceLength): message(message), numPiece(numPiece), pieceLength(pieceLength) {} virtual bool validate(Errors& error) { // TODO PeerMessageUtil::checkIndex(message->getIndex(), numPiece); PeerMessageUtil::checkBegin(message->getBegin(), pieceLength); return true; } }; typedef SharedHandle BtPieceMessageValidatorHandle; } // namespace aria2 #endif // _D_BT_PIECE_MESSAGE_VALIDATOR_H_