From: Matt Joiner <anacrolix@gmail.com>
Date: Thu, 10 Dec 2015 16:04:04 +0000 (+1100)
Subject: Don't consider a peer to have touched a piece if a chunk isn't written successfully
X-Git-Tag: v1.0.0~971
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=672e3c4136928f195b061d2e8e640d66f4d1069d;p=btrtrc.git

Don't consider a peer to have touched a piece if a chunk isn't written successfully

This prevents peers being banned when there are issues with the data service locally.
---

diff --git a/client.go b/client.go
index b2c9813c..c12eeb7d 100644
--- a/client.go
+++ b/client.go
@@ -2584,12 +2584,13 @@ func (me *Client) downloadedChunk(t *torrent, c *connection, msg *pp.Message) er
 			return
 		}
 		tr.Stop("write chunk")
+		me.mu.Lock()
+		if c.peerTouchedPieces == nil {
+			c.peerTouchedPieces = make(map[int]struct{})
+		}
+		c.peerTouchedPieces[int(req.Index)] = struct{}{}
+		me.mu.Unlock()
 	}()
-	// This could be made dependent on whether any actual data was written.
-	if c.peerTouchedPieces == nil {
-		c.peerTouchedPieces = make(map[int]struct{})
-	}
-	c.peerTouchedPieces[int(req.Index)] = struct{}{}
 
 	// log.Println("got chunk", req)
 	me.event.Broadcast()