From: Matt Joiner <anacrolix@gmail.com>
Date: Wed, 13 Oct 2021 08:05:19 +0000 (+1100)
Subject: Add expvar for successful peer wire protocol handshake peer reserved bytes
X-Git-Tag: v1.34.0^2~34
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=181cbd3e523d6b3918a754c7ae246c6ed0c1918c;p=btrtrc.git

Add expvar for successful peer wire protocol handshake peer reserved bytes
---

diff --git a/client.go b/client.go
index 83da43e6..3afe64a2 100644
--- a/client.go
+++ b/client.go
@@ -6,6 +6,7 @@ import (
 	"crypto/rand"
 	"encoding/binary"
 	"errors"
+	"expvar"
 	"fmt"
 	"io"
 	"net"
@@ -870,11 +871,20 @@ func (cl *Client) receiveHandshakes(c *PeerConn) (t *Torrent, err error) {
 	return
 }
 
+var successfulPeerWireProtocolHandshakePeerReservedBytes expvar.Map
+
+func init() {
+	torrent.Set(
+		"successful_peer_wire_protocol_handshake_peer_reserved_bytes",
+		&successfulPeerWireProtocolHandshakePeerReservedBytes)
+}
+
 func (cl *Client) connBtHandshake(c *PeerConn, ih *metainfo.Hash) (ret metainfo.Hash, err error) {
 	res, err := pp.Handshake(c.rw(), ih, cl.peerID, cl.config.Extensions)
 	if err != nil {
 		return
 	}
+	successfulPeerWireProtocolHandshakePeerReservedBytes.Add(res.PeerExtensionBits.String(), 1)
 	ret = res.Hash
 	c.PeerExtensionBytes = res.PeerExtensionBits
 	c.PeerID = res.PeerID