DHTNode.cc 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. /* <!-- copyright */
  2. /*
  3. * aria2 - The high speed download utility
  4. *
  5. * Copyright (C) 2006 Tatsuhiro Tsujikawa
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. *
  21. * In addition, as a special exception, the copyright holders give
  22. * permission to link the code of portions of this program with the
  23. * OpenSSL library under certain conditions as described in each
  24. * individual source file, and distribute linked combinations
  25. * including the two.
  26. * You must obey the GNU General Public License in all respects
  27. * for all of the code used other than OpenSSL. If you modify
  28. * file(s) with this exception, you may extend this exception to your
  29. * version of the file(s), but you are not obligated to do so. If you
  30. * do not wish to do so, delete this exception statement from your
  31. * version. If you delete this exception statement from all source
  32. * files in the program, then also delete it here.
  33. */
  34. /* copyright --> */
  35. #include "DHTNode.h"
  36. #include <cstring>
  37. #include "util.h"
  38. #include "a2functional.h"
  39. #include "bittorrent_helper.h"
  40. #include "wallclock.h"
  41. namespace aria2 {
  42. DHTNode::DHTNode()
  43. : port_(0), rtt_(0), condition_(0), lastContact_(Timer::zero())
  44. {
  45. generateID();
  46. }
  47. DHTNode::DHTNode(const unsigned char* id)
  48. : port_(0), rtt_(0), condition_(1), lastContact_(Timer::zero())
  49. {
  50. memcpy(id_, id, DHT_ID_LENGTH);
  51. }
  52. DHTNode::~DHTNode() {}
  53. void DHTNode::generateID() { util::generateRandomKey(id_); }
  54. bool DHTNode::operator==(const DHTNode& node) const
  55. {
  56. return memcmp(id_, node.id_, DHT_ID_LENGTH) == 0;
  57. }
  58. bool DHTNode::operator!=(const DHTNode& node) const { return !(*this == node); }
  59. bool DHTNode::operator<(const DHTNode& node) const
  60. {
  61. for (size_t i = 0; i < DHT_ID_LENGTH; ++i) {
  62. if (id_[i] > node.id_[i]) {
  63. return false;
  64. }
  65. else if (id_[i] < node.id_[i]) {
  66. return true;
  67. }
  68. }
  69. return true;
  70. }
  71. bool DHTNode::isGood() const { return !isBad() && !isQuestionable(); }
  72. constexpr int BAD_CONDITION = 5;
  73. bool DHTNode::isBad() const { return condition_ >= BAD_CONDITION; }
  74. bool DHTNode::isQuestionable() const
  75. {
  76. return !isBad() &&
  77. lastContact_.difference(global::wallclock()) >=
  78. DHT_NODE_CONTACT_INTERVAL;
  79. }
  80. void DHTNode::markGood() { condition_ = 0; }
  81. void DHTNode::markBad() { condition_ = BAD_CONDITION; }
  82. void DHTNode::updateLastContact() { lastContact_ = global::wallclock(); }
  83. void DHTNode::timeout() { ++condition_; }
  84. std::string DHTNode::toString() const
  85. {
  86. return fmt("DHTNode ID=%s, Host=%s(%u), Condition=%d, RTT=%ld",
  87. util::toHex(id_, DHT_ID_LENGTH).c_str(), ipaddr_.c_str(), port_,
  88. condition_, static_cast<long int>(rtt_.count()));
  89. }
  90. void DHTNode::setID(const unsigned char* id) { memcpy(id_, id, DHT_ID_LENGTH); }
  91. void DHTNode::setIPAddress(const std::string& ipaddr) { ipaddr_ = ipaddr; }
  92. } // namespace aria2