diff options
Diffstat (limited to 'src/BtMessageFactory.h')
| -rw-r--r-- | src/BtMessageFactory.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/BtMessageFactory.h b/src/BtMessageFactory.h index a7c55978..11ea7926 100644 --- a/src/BtMessageFactory.h +++ b/src/BtMessageFactory.h @@ -39,6 +39,9 @@ #include "BtMessage.h" #include "Piece.h" +class ExtensionMessage; +typedef SharedHandle<ExtensionMessage> ExensionMessageHandle; + class BtMessageFactory { public: virtual ~BtMessageFactory() {} @@ -84,6 +87,8 @@ public: createRejectMessage(int32_t index, int32_t begin, int32_t length) = 0; virtual BtMessageHandle createAllowedFastMessage(int32_t index) = 0; + + virtual BtMessageHandle createBtExtendedMessage(const ExensionMessageHandle& msg) = 0; }; typedef SharedHandle<BtMessageFactory> BtMessageFactoryHandle; |