Don't download piece which is filtered out (e.g., --select-file)

This commit fixes the bug that piece which is not requested by
--select-file is downloaded in multi-file WebSeeding.

1 files changed, 6 insertions, 1 deletions

diff --git a/src/DefaultPieceStorage.cc b/src/DefaultPieceStorage.cc
index 7e94ebc6..456058a5 100644
--- a/src/DefaultPieceStorage.cc
+++ b/src/DefaultPieceStorage.cc
@@ -119,6 +119,9 @@ DefaultPieceStorage::~DefaultPieceStorage() = default;
 std::shared_ptr<Piece> DefaultPieceStorage::checkOutPiece(size_t index,
                                                           cuid_t cuid)
 {
+  assert(!bitfieldMan_->isFilterEnabled() ||
+         bitfieldMan_->isFilterBitSet(index));
+
   bitfieldMan_->setUseBit(index);
 
   std::shared_ptr<Piece> piece = findUsedPiece(index);
@@ -398,7 +401,9 @@ DefaultPieceStorage::getMissingPiece(size_t minSplitSize,
 std::shared_ptr<Piece> DefaultPieceStorage::getMissingPiece(size_t index,
                                                             cuid_t cuid)
 {
-  if (hasPiece(index) || isPieceUsed(index)) {
+  if (hasPiece(index) || isPieceUsed(index) ||
+      (bitfieldMan_->isFilterEnabled() &&
+       !bitfieldMan_->isFilterBitSet(index))) {
     return nullptr;
   }
   else {