Wrap Command object in std::unique_ptr

1 files changed, 6 insertions, 3 deletions

diff --git a/src/FileAllocationEntry.h b/src/FileAllocationEntry.h
index 74a455d5..6811f56f 100644
--- a/src/FileAllocationEntry.h
+++ b/src/FileAllocationEntry.h
@@ -52,7 +52,9 @@ class FileAllocationEntry : public RequestGroupEntry, public ProgressAwareEntry
 private:
   std::shared_ptr<FileAllocationIterator> fileAllocationIterator_;
 public:
-  FileAllocationEntry(RequestGroup* requestGroup, Command* nextCommand = 0);
+  FileAllocationEntry(RequestGroup* requestGroup,
+                      std::unique_ptr<Command> nextCommand =
+                      std::unique_ptr<Command>());
 
   ~FileAllocationEntry();
 
@@ -64,8 +66,9 @@ public:
 
   void allocateChunk();
 
-  virtual void prepareForNextAction(std::vector<Command*>& commands,
-                                    DownloadEngine* e) = 0;
+  virtual void prepareForNextAction
+  (std::vector<std::unique_ptr<Command>>& commands,
+   DownloadEngine* e) = 0;
 };
 
 } // namespace aria2