Record connection dirtied piece before queuing for hash

Not a bug, but probably more robust.

diff --git a/connection.go b/connection.go
index 3a857d91e123808a750e09c24c15fd068d0cee08..76c15d3262fd60bed6f626e04719b9b4f5360c5e 100644 (file)
--- a/connection.go
+++ b/connection.go
@@ -1400,13 +1400,13 @@ func (c *connection) receiveChunk(msg *pp.Message) error {
                return nil
        }
 
+       c.onDirtiedPiece(pieceIndex(req.Index))
+
        if t.pieceAllDirty(pieceIndex(req.Index)) {
                t.queuePieceCheck(pieceIndex(req.Index))
                t.pendAllChunkSpecs(pieceIndex(req.Index))
        }
 
-       c.onDirtiedPiece(pieceIndex(req.Index))
-
        cl.event.Broadcast()
        // We do this because we've written a chunk, and may change PieceState.Partial.
        t.publishPieceChange(pieceIndex(req.Index))