Managed user cuid in Piece.

Previous implementation does not store information which Command cuid
uses which Piece.  Mark Piece acquired by SegmentMan by setting
Piece::setUsedBySegment(true).

1 files changed, 7 insertions, 0 deletions

diff --git a/src/UTMetadataRequestFactory.h b/src/UTMetadataRequestFactory.h
index e8e02b2a..7af3e8da 100644
--- a/src/UTMetadataRequestFactory.h
+++ b/src/UTMetadataRequestFactory.h
@@ -40,6 +40,7 @@
 #include <vector>
 
 #include "SharedHandle.h"
+#include "Command.h"
 
 namespace aria2 {
 
@@ -62,6 +63,7 @@ private:
   BtMessageFactory* messageFactory_;
 
   UTMetadataRequestTracker* tracker_;
+  cuid_t cuid_;
 public:
   UTMetadataRequestFactory();
 
@@ -94,6 +96,11 @@ public:
   {
     tracker_ = tracker;
   }
+
+  void setCuid(cuid_t cuid)
+  {
+    cuid_ = cuid;
+  }
 };
 
 } // namespace aria2