peer: Track and return advertised protocol version

This adds a new field to the peer struct which stores the protocol
version advertised by the remote peer and updates the StatsSnapshot to
return the advertised version instead of the negotiated version.
This commit is contained in:
Dave Collins 2016-10-27 18:40:06 -05:00
parent 14b51fc5f8
commit a1f014c9e1
No known key found for this signature in database
GPG key ID: B8904D9D9C93D1F2

View file

@ -402,7 +402,8 @@ type Peer struct {
userAgent string
services wire.ServiceFlag
versionKnown bool
protocolVersion uint32
advertisedProtoVer uint32 // protocol version advertised by remote
protocolVersion uint32 // negotiated protocol version
sendHeadersPreferred bool // peer sent a sendheaders message
versionSent bool
verAckReceived bool
@ -488,7 +489,7 @@ func (p *Peer) StatsSnapshot() *StatsSnap {
addr := p.addr
userAgent := p.userAgent
services := p.services
protocolVersion := p.protocolVersion
protocolVersion := p.advertisedProtoVer
p.flagsMtx.Unlock()
// Get a copy of all relevant flags and stats.
@ -644,7 +645,7 @@ func (p *Peer) VerAckReceived() bool {
return verAckReceived
}
// ProtocolVersion returns the peer protocol version.
// ProtocolVersion returns the negotiated peer protocol version.
//
// This function is safe for concurrent access.
func (p *Peer) ProtocolVersion() uint32 {
@ -1029,7 +1030,8 @@ func (p *Peer) handleRemoteVersionMsg(msg *wire.MsgVersion) error {
// Negotiate the protocol version.
p.flagsMtx.Lock()
p.protocolVersion = minUint32(p.protocolVersion, uint32(msg.ProtocolVersion))
p.advertisedProtoVer = uint32(msg.ProtocolVersion)
p.protocolVersion = minUint32(p.protocolVersion, p.advertisedProtoVer)
p.versionKnown = true
log.Debugf("Negotiated protocol version %d for peer %s",
p.protocolVersion, p)