1 files changed, 5 insertions, 0 deletions
diff --git a/src/DefaultExtensionMessageFactory.h b/src/DefaultExtensionMessageFactory.h
index 6bbc45ef..e1cef02f 100644
--- a/src/DefaultExtensionMessageFactory.h
+++ b/src/DefaultExtensionMessageFactory.h
@@ -40,6 +40,7 @@
 namespace aria2 {
 
 class BtContext;
+class PeerStorage;
 class Peer;
 class Logger;
 
@@ -47,6 +48,8 @@ class DefaultExtensionMessageFactory:public ExtensionMessageFactory {
 private:
   SharedHandle<BtContext> _btContext;
 
+  SharedHandle<PeerStorage> _peerStorage;
+
   SharedHandle<Peer> _peer;
 
   Logger* _logger;
@@ -64,6 +67,8 @@ public:
 
   void setBtContext(const SharedHandle<BtContext>& btContext);
 
+  void setPeerStorage(const SharedHandle<PeerStorage>& peerStorage);
+
   void setPeer(const SharedHandle<Peer>& peer);
 };