summaryrefslogtreecommitdiff
path: root/src/DHTAnnouncePeerMessage.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/DHTAnnouncePeerMessage.cc')
-rw-r--r--src/DHTAnnouncePeerMessage.cc56
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