Merge pull request aria2#1065 from kwkam/unknown-stat

UnknownLengthPieceStorage: Completed Length
hhlass · Nov 8, 2017 · 5e7976a · 5e7976a
2 parents 45756dd + eef212d
commit 5e7976a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/UnknownLengthPieceStorage.cc b/src/UnknownLengthPieceStorage.cc
@@ -210,6 +210,15 @@ bool UnknownLengthPieceStorage::isPieceUsed(size_t index)
   }
 }
 
+int64_t UnknownLengthPieceStorage::getCompletedLength()
+{
+  // TODO we have to return actual completed length here?
+  if (piece_) {
+    return piece_->getLength();
+  }
+  return totalLength_;
+}
+
 std::shared_ptr<DiskAdaptor> UnknownLengthPieceStorage::getDiskAdaptor()
 {
   return diskAdaptor_;

diff --git a/src/UnknownLengthPieceStorage.h b/src/UnknownLengthPieceStorage.h
@@ -162,11 +162,7 @@ class UnknownLengthPieceStorage : public PieceStorage {
     return totalLength_;
   }
 
-  virtual int64_t getCompletedLength() CXX11_OVERRIDE
-  {
-    // TODO we have to return actual completed length here?
-    return totalLength_;
-  }
+  virtual int64_t getCompletedLength() CXX11_OVERRIDE;
 
   virtual int64_t getFilteredCompletedLength() CXX11_OVERRIDE
   {