From 529b9fdceb7b5a277a983badd29dca3cc69259c5 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Mon, 1 Jul 2013 21:42:51 +0900
Subject: Use std::unique_ptr for ExtensionMessage

---
 src/UTMetadataRequestExtensionMessage.h | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

(limited to 'src/UTMetadataRequestExtensionMessage.h')

diff --git a/src/UTMetadataRequestExtensionMessage.h b/src/UTMetadataRequestExtensionMessage.h
index f7896f13..eaba6b3c 100644
--- a/src/UTMetadataRequestExtensionMessage.h
+++ b/src/UTMetadataRequestExtensionMessage.h
@@ -48,7 +48,7 @@ class Peer;
 
 class UTMetadataRequestExtensionMessage:public UTMetadataExtensionMessage {
 private:
-  std::shared_ptr<DownloadContext> dctx_;
+  DownloadContext* dctx_;
 
   std::shared_ptr<Peer> peer_;
 
@@ -58,25 +58,17 @@ private:
 public:
   UTMetadataRequestExtensionMessage(uint8_t extensionMessageID);
 
-  ~UTMetadataRequestExtensionMessage();
-
   virtual std::string getPayload();
 
   virtual std::string toString() const;
 
   virtual void doReceivedAction();
 
-  void setDownloadContext(const std::shared_ptr<DownloadContext>& dctx);
+  void setDownloadContext(DownloadContext* dctx);
 
-  void setBtMessageDispatcher(BtMessageDispatcher* disp)
-  {
-    dispatcher_ = disp;
-  }
+  void setBtMessageDispatcher(BtMessageDispatcher* disp);
 
-  void setBtMessageFactory(BtMessageFactory* factory)
-  {
-    messageFactory_ = factory;
-  }
+  void setBtMessageFactory(BtMessageFactory* factory);
 
   void setPeer(const std::shared_ptr<Peer>& peer);
 };
-- 
cgit