1 files changed, 7 insertions, 0 deletions
diff --git a/src/PeerSessionResource.h b/src/PeerSessionResource.h
index c25eb9fb..f0828c0e 100644
--- a/src/PeerSessionResource.h
+++ b/src/PeerSessionResource.h
@@ -45,6 +45,7 @@
 namespace aria2 {
 
 class BitfieldMan;
+class BtMessageDispatcher;
 
 class PeerSessionResource {
 private:
@@ -78,6 +79,8 @@ private:
   Time _lastDownloadUpdate;
 
   Time _lastAmUnchoking;
+
+  WeakHandle<BtMessageDispatcher> _dispatcher;
 public:
   PeerSessionResource(size_t pieceLength, uint64_t totalLength);
 
@@ -185,6 +188,10 @@ public:
   const Time& getLastAmUnchoking() const;
 
   uint64_t getCompletedLength() const;
+
+  void setBtMessageDispatcher(const WeakHandle<BtMessageDispatcher>& dpt);
+
+  size_t countOutstandingUpload() const;
 };
 
 } // namespace aria2