From aa3478ae2a5987da956bf6f39a7e4c7839fd8055 Mon Sep 17 00:00:00 2001
From: dvovk <vovk.dimon@gmail.com>
Date: Thu, 22 Aug 2024 13:56:33 +0000
Subject: [PATCH] fixed completed bytes count

---
 t.go       |  3 ---
 torrent.go | 12 +++++++-----
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/t.go b/t.go
index 3add0ae9d8..92f36485ed 100644
--- a/t.go
+++ b/t.go
@@ -256,9 +256,6 @@ func (t *Torrent) DownloadPieces(begin, end pieceIndex) {
 						cs.pieceHashed(size)
 					})
 					storage.MarkComplete(false)
-					t.allStats(func(cs *ConnStats) {
-						cs.pieceCompleted(size)
-					})
 					t.updatePieceCompletion(piece.index, true)
 					return nil
 				}
diff --git a/torrent.go b/torrent.go
index 5516ae28ea..fa034ebe6e 100644
--- a/torrent.go
+++ b/torrent.go
@@ -1770,6 +1770,12 @@ func (t *Torrent) updatePieceCompletion(piece pieceIndex, lock bool) bool {
 	x := uint32(piece)
 
 	if complete {
+		if !t._completedPieces.Contains(x) {
+			t.allStats(func(cs *ConnStats) {
+				cs.pieceCompleted(p.Info().Length())
+			})
+		}
+
 		t._completedPieces.Add(x)
 		t.openNewConns(false)
 	} else {
@@ -2645,11 +2651,7 @@ func (t *Torrent) pieceHashed(piece pieceIndex, passed bool, hashIoErr error) {
 
 		err := p.Storage().MarkComplete(!hasDirtyChunks)
 
-		if err == nil {
-			t.allStats(func(cs *ConnStats) {
-				cs.pieceCompleted(int64(p.length(true)))
-			})
-		} else {
+		if err != nil {
 			t.logger.Levelf(log.Warning, "%T: error marking piece complete %d: %s", t.storage, piece, err)
 		}