1 files changed, 24 insertions, 0 deletions
diff --git a/src/DefaultExtensionMessageFactory.h b/src/DefaultExtensionMessageFactory.h
index dfc1c0ec..319528f6 100644
--- a/src/DefaultExtensionMessageFactory.h
+++ b/src/DefaultExtensionMessageFactory.h
@@ -43,6 +43,9 @@ class PeerStorage;
 class Peer;
 class Logger;
 class ExtensionMessageRegistry;
+class DownloadContext;
+class BtMessageFactory;
+class BtMessageDispatcher;
 
 class DefaultExtensionMessageFactory:public ExtensionMessageFactory {
 private:
@@ -52,6 +55,12 @@ private:
 
   SharedHandle<ExtensionMessageRegistry> _registry;
 
+  SharedHandle<DownloadContext> _dctx;
+
+  WeakHandle<BtMessageFactory> _messageFactory;
+
+  WeakHandle<BtMessageDispatcher> _dispatcher;
+
   Logger* _logger;
 
 public:
@@ -75,6 +84,21 @@ public:
   {
     _registry = registry;
   }
+
+  void setDownloadContext(const SharedHandle<DownloadContext>& dctx)
+  {
+    _dctx = dctx;
+  }
+
+  void setBtMessageFactory(const WeakHandle<BtMessageFactory>& factory)
+  {
+    _messageFactory = factory;
+  }
+
+  void setBtMessageDispatcher(const WeakHandle<BtMessageDispatcher>& disp)
+  {
+    _dispatcher = disp;
+  }
 };
 
 typedef SharedHandle<DefaultExtensionMessageFactory> DefaultExtensionMessageFactoryHandle;