diff options
Diffstat (limited to 'src/DHTAnnouncePeerMessage.cc')
| -rw-r--r-- | src/DHTAnnouncePeerMessage.cc | 56 |
1 files changed, 25 insertions, 31 deletions
diff --git a/src/DHTAnnouncePeerMessage.cc b/src/DHTAnnouncePeerMessage.cc index ef4c84fe..3ba73dc3 100644 --- a/src/DHTAnnouncePeerMessage.cc +++ b/src/DHTAnnouncePeerMessage.cc @@ -60,30 +60,28 @@ const std::string DHTAnnouncePeerMessage::PORT("port"); const std::string DHTAnnouncePeerMessage::TOKEN("token"); -DHTAnnouncePeerMessage::DHTAnnouncePeerMessage -(const std::shared_ptr<DHTNode>& localNode, - const std::shared_ptr<DHTNode>& remoteNode, - const unsigned char* infoHash, - uint16_t tcpPort, - const std::string& token, - const std::string& transactionID) - : DHTQueryMessage{localNode, remoteNode, transactionID}, - token_{token}, - tcpPort_{tcpPort}, - peerAnnounceStorage_{nullptr}, - tokenTracker_{nullptr} +DHTAnnouncePeerMessage::DHTAnnouncePeerMessage( + const std::shared_ptr<DHTNode>& localNode, + const std::shared_ptr<DHTNode>& remoteNode, const unsigned char* infoHash, + uint16_t tcpPort, const std::string& token, + const std::string& transactionID) + : DHTQueryMessage{localNode, remoteNode, transactionID}, + token_{token}, + tcpPort_{tcpPort}, + peerAnnounceStorage_{nullptr}, + tokenTracker_{nullptr} { memcpy(infoHash_, infoHash, DHT_ID_LENGTH); } void DHTAnnouncePeerMessage::doReceivedAction() { - peerAnnounceStorage_->addPeerAnnounce - (infoHash_, getRemoteNode()->getIPAddress(), tcpPort_); + peerAnnounceStorage_->addPeerAnnounce( + infoHash_, getRemoteNode()->getIPAddress(), tcpPort_); - getMessageDispatcher()->addMessageToQueue - (getMessageFactory()->createAnnouncePeerReplyMessage - (getRemoteNode(), getTransactionID())); + getMessageDispatcher()->addMessageToQueue( + getMessageFactory()->createAnnouncePeerReplyMessage(getRemoteNode(), + getTransactionID())); } std::unique_ptr<Dict> DHTAnnouncePeerMessage::getArgument() @@ -103,19 +101,17 @@ const std::string& DHTAnnouncePeerMessage::getMessageType() const void DHTAnnouncePeerMessage::validate() const { - if(!tokenTracker_->validateToken(token_, infoHash_, - getRemoteNode()->getIPAddress(), - getRemoteNode()->getPort())) { - throw DL_ABORT_EX - (fmt("Invalid token=%s from %s:%u", - util::toHex(token_).c_str(), - getRemoteNode()->getIPAddress().c_str(), - getRemoteNode()->getPort())); + if (!tokenTracker_->validateToken(token_, infoHash_, + getRemoteNode()->getIPAddress(), + getRemoteNode()->getPort())) { + throw DL_ABORT_EX(fmt( + "Invalid token=%s from %s:%u", util::toHex(token_).c_str(), + getRemoteNode()->getIPAddress().c_str(), getRemoteNode()->getPort())); } } -void DHTAnnouncePeerMessage::setPeerAnnounceStorage -(DHTPeerAnnounceStorage* storage) +void DHTAnnouncePeerMessage::setPeerAnnounceStorage( + DHTPeerAnnounceStorage* storage) { peerAnnounceStorage_ = storage; } @@ -127,10 +123,8 @@ void DHTAnnouncePeerMessage::setTokenTracker(DHTTokenTracker* tokenTracker) std::string DHTAnnouncePeerMessage::toStringOptional() const { - return fmt("token=%s, info_hash=%s, tcpPort=%u", - util::toHex(token_).c_str(), - util::toHex(infoHash_, INFO_HASH_LENGTH).c_str(), - tcpPort_); + return fmt("token=%s, info_hash=%s, tcpPort=%u", util::toHex(token_).c_str(), + util::toHex(infoHash_, INFO_HASH_LENGTH).c_str(), tcpPort_); } } // namespace aria2 |
