Count outgoing extended messages by extension protocol

diff --git a/peerconn.go b/peerconn.go
index a5153fbfb1999e257691768616e4c160ba15f388..1be1289192165dd095f94850605f5f28d8212acc 100644 (file)
--- a/peerconn.go
+++ b/peerconn.go
@@ -863,6 +863,14 @@ func (c *PeerConn) requestPendingMetadata() {
 
 func (cn *PeerConn) wroteMsg(msg *pp.Message) {
        torrent.Add(fmt.Sprintf("messages written of type %s", msg.Type.String()), 1)
+       if msg.Type == pp.Extended {
+               for name, id := range cn.PeerExtensionIDs {
+                       if id != msg.ExtendedID {
+                               continue
+                       }
+                       torrent.Add(fmt.Sprintf("Extended messages written for protocol %q", name), 1)
+               }
+       }
        cn.allStats(func(cs *ConnStats) { cs.wroteMsg(msg) })
 }