Use std::unique_ptr for ExtensionMessage

1 files changed, 28 insertions, 8 deletions

diff --git a/src/UTMetadataDataExtensionMessage.cc b/src/UTMetadataDataExtensionMessage.cc
index 88d78d71..208749c0 100644
--- a/src/UTMetadataDataExtensionMessage.cc
+++ b/src/UTMetadataDataExtensionMessage.cc
@@ -54,13 +54,13 @@ namespace aria2 {
 
 UTMetadataDataExtensionMessage::UTMetadataDataExtensionMessage
 (uint8_t extensionMessageID)
-  : UTMetadataExtensionMessage(extensionMessageID),
-    totalSize_(0),
-    tracker_(0)
+  : UTMetadataExtensionMessage{extensionMessageID},
+    totalSize_{0},
+    dctx_{nullptr},
+    pieceStorage_{nullptr},
+    tracker_{nullptr}
 {}
 
-UTMetadataDataExtensionMessage::~UTMetadataDataExtensionMessage() {}
-
 std::string UTMetadataDataExtensionMessage::getPayload()
 {
   Dict dict;
@@ -109,19 +109,39 @@ void UTMetadataDataExtensionMessage::doReceivedAction()
   }
 }
 
+void UTMetadataDataExtensionMessage::setTotalSize(size_t totalSize)
+{
+  totalSize_ = totalSize;
+}
+
+size_t UTMetadataDataExtensionMessage::getTotalSize() const
+{
+  return totalSize_;
+}
+
 void UTMetadataDataExtensionMessage::setData(const std::string& data)
 {
   data_ = data;
 }
 
+const std::string& UTMetadataDataExtensionMessage::getData() const
+{
+  return data_;
+}
+
 void UTMetadataDataExtensionMessage::setPieceStorage
-(const std::shared_ptr<PieceStorage>& pieceStorage)
+(PieceStorage* pieceStorage)
 {
   pieceStorage_ = pieceStorage;
 }
 
-void UTMetadataDataExtensionMessage::setDownloadContext
-(const std::shared_ptr<DownloadContext>& dctx)
+void UTMetadataDataExtensionMessage::setUTMetadataRequestTracker
+(UTMetadataRequestTracker* tracker)
+{
+  tracker_ = tracker;
+}
+
+void UTMetadataDataExtensionMessage::setDownloadContext(DownloadContext* dctx)
 {
   dctx_ = dctx;
 }