Count requested chunk lengths

Have seen some very ridiculous lengths causing data read errors

diff --git a/connection.go b/connection.go
index 46083d56d3794b6aa2d55a641a69a72951d2f238..fbf86e56600cb993db8fae4f85a1cf864e5c391b 100644 (file)
--- a/connection.go
+++ b/connection.go
@@ -884,6 +884,7 @@ func (c *connection) mainReadLoop() error {
                case pp.Have:
                        err = c.peerSentHave(int(msg.Index))
                case pp.Request:
+                       requestedChunkLengths.Add(strconv.FormatUint(msg.Length.Uint64(), 10), 1)
                        if c.Choked {
                                break
                        }

diff --git a/global.go b/global.go
index ffc30f493348c455c360187964c935d66a0899b8..4b27f2e50b7a2e277e57a2a28157ca12c7c7332d 100644 (file)
--- a/global.go
+++ b/global.go
@@ -62,6 +62,7 @@ var (
        postedKeepalives           = expvar.NewInt("postedKeepalives")
        // Requests received for pieces we don't have.
        requestsReceivedForMissingPieces = expvar.NewInt("requestsReceivedForMissingPieces")
+       requestedChunkLengths            = expvar.NewMap("requestedChunkLengths")
 
        messageTypesReceived = expvar.NewMap("messageTypesReceived")
        messageTypesSent     = expvar.NewMap("messageTypesSent")

diff --git a/peer_protocol/protocol.go b/peer_protocol/protocol.go
index bc4d6ee604451711f812e37129b4d488d4bf3325..27827eed72afa693604b78c3137b2bee77e30f2b 100644 (file)
--- a/peer_protocol/protocol.go
+++ b/peer_protocol/protocol.go
@@ -25,6 +25,10 @@ func (i Integer) Int() int {
        return int(i)
 }
 
+func (i Integer) Uint64() uint64 {
+       return uint64(i)
+}
+
 const (
        Protocol = "\x13BitTorrent protocol"
 )