Show bitfield for unknown length download in aria2.tellStatus RPC method

Generally, bitfield is not available for download whose total length
is unknown.  We create bitfield when download is completed (usually
connection EOF) so that we can use it to show additional info in RPC
aria2.tellStatus response.  Specifically, bitfield is now shown.  And
completedLength under files key (or completedLength in
aria2.getFiles() response) is correctly shown.

1 files changed, 8 insertions, 8 deletions

diff --git a/src/UnknownLengthPieceStorage.h b/src/UnknownLengthPieceStorage.h
index 5aad9667..d39f3cae 100644
--- a/src/UnknownLengthPieceStorage.h
+++ b/src/UnknownLengthPieceStorage.h
@@ -43,6 +43,7 @@ class Option;
 class DownloadContext;
 class DiskWriterFactory;
 class DirectDiskAdaptor;
+class BitfieldMan;
 
 class UnknownLengthPieceStorage:public PieceStorage {
 private:
@@ -54,9 +55,14 @@ private:
 
   int64_t totalLength_;
 
+  std::unique_ptr<BitfieldMan> bitfield_;
+
   bool downloadFinished_;
 
   std::shared_ptr<Piece> piece_;
+
+  void createBitfield();
+
 public:
   UnknownLengthPieceStorage(const std::shared_ptr<DownloadContext>& downloadContext);
 
@@ -203,18 +209,12 @@ public:
    */
   virtual void initStorage() CXX11_OVERRIDE;
 
-  virtual const unsigned char* getBitfield() CXX11_OVERRIDE
-  {
-    return nullptr;
-  }
+  virtual const unsigned char* getBitfield() CXX11_OVERRIDE;
 
   virtual void setBitfield(const unsigned char* bitfield,
                            size_t bitfieldLength) CXX11_OVERRIDE {}
 
-  virtual size_t getBitfieldLength() CXX11_OVERRIDE
-  {
-    return 0;
-  }
+  virtual size_t getBitfieldLength() CXX11_OVERRIDE;
 
   virtual bool isSelectiveDownloadingMode() CXX11_OVERRIDE
   {