From 656dc8cc0034d3dd57b5e1baecb788d9c4e9929d Mon Sep 17 00:00:00 2001
From: Justin Traglia <95511699+jtraglia@users.noreply.github.com>
Date: Fri, 19 Aug 2022 01:02:47 -0500
Subject: [PATCH] eth, les: unlock downloader peerSet if there's an error
 (#25546)

Unlock peerSet if there's an error in the downloader
---
 eth/downloader/peer.go | 1 +
 les/downloader/peer.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/eth/downloader/peer.go b/eth/downloader/peer.go
index d74d23e74d557..6b8269495948e 100644
--- a/eth/downloader/peer.go
+++ b/eth/downloader/peer.go
@@ -237,6 +237,7 @@ func (ps *peerSet) Register(p *peerConnection) error {
 	}
 	p.rates = msgrate.NewTracker(ps.rates.MeanCapacities(), ps.rates.MedianRoundTrip())
 	if err := ps.rates.Track(p.id, p.rates); err != nil {
+		ps.lock.Unlock()
 		return err
 	}
 	ps.peers[p.id] = p
diff --git a/les/downloader/peer.go b/les/downloader/peer.go
index 5a92e9cf9b87a..c2161e2dae425 100644
--- a/les/downloader/peer.go
+++ b/les/downloader/peer.go
@@ -350,6 +350,7 @@ func (ps *peerSet) Register(p *peerConnection) error {
 	}
 	p.rates = msgrate.NewTracker(ps.rates.MeanCapacities(), ps.rates.MedianRoundTrip())
 	if err := ps.rates.Track(p.id, p.rates); err != nil {
+		ps.lock.Unlock()
 		return err
 	}
 	ps.peers[p.id] = p