diff options
Diffstat (limited to 'src/PeerInteractionCommand.cc')
| -rw-r--r-- | src/PeerInteractionCommand.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/PeerInteractionCommand.cc b/src/PeerInteractionCommand.cc index 3fe2035f..7507be30 100644 --- a/src/PeerInteractionCommand.cc +++ b/src/PeerInteractionCommand.cc @@ -109,6 +109,8 @@ PeerInteractionCommand::PeerInteractionCommand SharedHandle<DefaultExtensionMessageFactory> extensionMessageFactory (new DefaultExtensionMessageFactory(peer, exMsgRegistry)); extensionMessageFactory->setPeerStorage(peerStorage); + extensionMessageFactory->setDownloadContext + (_requestGroup->getDownloadContext()); SharedHandle<DefaultBtMessageFactory> factory(new DefaultBtMessageFactory()); factory->setCuid(cuid); @@ -189,6 +191,9 @@ PeerInteractionCommand::PeerInteractionCommand factory->setBtRequestFactory(reqFactory); factory->setPeerConnection(peerConnection); + extensionMessageFactory->setBtMessageDispatcher(dispatcher); + extensionMessageFactory->setBtMessageFactory(factory); + peer->allocateSessionResource (_requestGroup->getDownloadContext()->getPieceLength(), _requestGroup->getDownloadContext()->getTotalLength()); |