summaryrefslogtreecommitdiff
path: root/src/PeerSessionResource.cc
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2008-11-03 07:16:25 +0000
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2008-11-03 07:16:25 +0000
commit7818e0e770f2f89d3f2356d887cf3d4559ddd7c3 (patch)
tree11f1cad244028df7f27e0c86963c67de0327c7f7 /src/PeerSessionResource.cc
parent7cf589e78483e6ca7daff2ddd1a30af8091a4448 (diff)
2008-11-03 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
Implemented commented code in BtSeederStateChoke * src/BtSeederStateChoke.cc * src/Peer.cc * src/Peer.h * src/PeerInteractionCommand.cc * src/PeerSessionResource.cc * src/PeerSessionResource.h * test/MockBtMessageDispatcher.h * test/PeerSessionResourceTest.cc
Diffstat (limited to 'src/PeerSessionResource.cc')
-rw-r--r--src/PeerSessionResource.cc18
1 files changed, 17 insertions, 1 deletions
diff --git a/src/PeerSessionResource.cc b/src/PeerSessionResource.cc
index c6436a3c..a35ceadb 100644
--- a/src/PeerSessionResource.cc
+++ b/src/PeerSessionResource.cc
@@ -33,10 +33,14 @@
*/
/* copyright --> */
#include "PeerSessionResource.h"
+
+#include <cassert>
+#include <algorithm>
+
#include "BitfieldManFactory.h"
#include "BitfieldMan.h"
#include "A2STR.h"
-#include <algorithm>
+#include "BtMessageDispatcher.h"
namespace aria2 {
@@ -338,4 +342,16 @@ uint64_t PeerSessionResource::getCompletedLength() const
return _bitfieldMan->getCompletedLength();
}
+void PeerSessionResource::setBtMessageDispatcher
+(const WeakHandle<BtMessageDispatcher>& dpt)
+{
+ _dispatcher = dpt;
+}
+
+size_t PeerSessionResource::countOutstandingUpload() const
+{
+ assert(!_dispatcher.isNull());
+ return _dispatcher->countOutstandingUpload();
+}
+
} // namespace aria2