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/DefaultBtMessageFactory.cc | |
| 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/DefaultBtMessageFactory.cc')
| -rw-r--r-- | src/DefaultBtMessageFactory.cc | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/DefaultBtMessageFactory.cc b/src/DefaultBtMessageFactory.cc index 6cba336f..e480ecad 100644 --- a/src/DefaultBtMessageFactory.cc +++ b/src/DefaultBtMessageFactory.cc @@ -96,7 +96,12 @@ DefaultBtMessageFactory::createBtMessage(const unsigned char* data, size_t dataL msg = BtUnchokeMessage::create(data, dataLength); break; case BtInterestedMessage::ID: - msg = BtInterestedMessage::create(data, dataLength); + { + SharedHandle<BtInterestedMessage> m = + BtInterestedMessage::create(data, dataLength); + m->setPeerStorage(_peerStorage); + msg = m; + } break; case BtNotInterestedMessage::ID: { |