Use std::unique_ptr for ExtensionMessage

1 files changed, 3 insertions, 3 deletions

diff --git a/src/UTMetadataRequestFactory.h b/src/UTMetadataRequestFactory.h
index e8de1e67..7e5688d8 100644
--- a/src/UTMetadataRequestFactory.h
+++ b/src/UTMetadataRequestFactory.h
@@ -54,7 +54,7 @@ class BtMessage;
 
 class UTMetadataRequestFactory {
 private:
-  std::shared_ptr<DownloadContext> dctx_;
+  DownloadContext* dctx_;
 
   std::shared_ptr<Peer> peer_;
 
@@ -70,9 +70,9 @@ public:
   // Creates and returns at most num of ut_metadata request
   // message. pieceStorage is used to identify missing piece.
   std::vector<std::unique_ptr<BtMessage>> create
-  (size_t num, const std::shared_ptr<PieceStorage>& pieceStorage);
+  (size_t num, PieceStorage* pieceStorage);
 
-  void setDownloadContext(const std::shared_ptr<DownloadContext>& dctx)
+  void setDownloadContext(DownloadContext* dctx)
   {
     dctx_ = dctx;
   }