diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2008-11-03 12:04:57 +0000 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2008-11-03 12:04:57 +0000 |
| commit | 6d01f8f94ff83241fbae5130d1904e03a08945aa (patch) | |
| tree | bc844a75513ea5b8c89a6aba455d4e75818e25ec /src/BtInterestedMessage.h | |
| parent | 8fab0ff2160399487fb0010d3b0c8b431dfe545f (diff) | |
2008-11-03 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
Execute choking algorithm when BtInterestedMessage arrives from
unchoked peer.
* src/BtInterestedMessage.cc
* src/BtInterestedMessage.h
* src/DefaultBtMessageFactory.cc
* test/BtInterestedMessageTest.cc
* test/MockPeerStorage.h
Diffstat (limited to 'src/BtInterestedMessage.h')
| -rw-r--r-- | src/BtInterestedMessage.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/BtInterestedMessage.h b/src/BtInterestedMessage.h index 9088b259..08727f09 100644 --- a/src/BtInterestedMessage.h +++ b/src/BtInterestedMessage.h @@ -39,6 +39,7 @@ namespace aria2 { +class PeerStorage; class BtInterestedMessage; typedef SharedHandle<BtInterestedMessage> BtInterestedMessageHandle; @@ -47,6 +48,8 @@ class BtInterestedMessage : public SimpleBtMessage { private: unsigned char* msg; + SharedHandle<PeerStorage> _peerStorage; + static size_t MESSAGE_LENGTH; public: @@ -73,6 +76,8 @@ public: virtual bool sendPredicate() const; virtual void onSendComplete(); + + void setPeerStorage(const SharedHandle<PeerStorage>& peerStorage); }; } // namespace aria2 |