/* */ #ifndef D_BT_HANDSHAKE_MESSAGE_H #define D_BT_HANDSHAKE_MESSAGE_H #include "SimpleBtMessage.h" namespace aria2 { class BtHandshakeMessage : public SimpleBtMessage { public: static const size_t PSTR_LENGTH = 19; static const unsigned char* BT_PSTR; static const size_t RESERVED_LENGTH = 8; static const size_t MESSAGE_LENGTH = 68; private: uint8_t pstrlen_; unsigned char* pstr_; unsigned char* reserved_; unsigned char* infoHash_; unsigned char* peerId_; void init(); public: BtHandshakeMessage(); /** * infoHash must be 20 byte length. * peerId must be 20 byte length. */ BtHandshakeMessage(const unsigned char* infoHash, const unsigned char* peerId); static SharedHandle create(const unsigned char* data, size_t dataLength); virtual ~BtHandshakeMessage() { delete [] pstr_; delete [] reserved_; delete [] infoHash_; delete [] peerId_; } static const uint8_t ID = INT8_MAX; static const std::string NAME; virtual void doReceivedAction() {}; virtual unsigned char* createMessage(); virtual size_t getMessageLength(); virtual std::string toString() const; bool isFastExtensionSupported() const; bool isExtendedMessagingEnabled() const; bool isDHTEnabled() const; void setDHTEnabled(bool enabled) { if(enabled) { reserved_[7] |= 0x01u; } else { reserved_[7] &= ~0x01u; } } uint8_t getPstrlen() const { return pstrlen_; } const unsigned char* getPstr() const { return pstr_; } const unsigned char* getReserved() const { return reserved_; } const unsigned char* getInfoHash() const { return infoHash_; } void setInfoHash(const unsigned char* infoHash); const unsigned char* getPeerId() const { return peerId_; } void setPeerId(const unsigned char* peerId); }; } // namespace aria2 #endif // D_HANDSHAKE_MESSAGE_H