mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-18 19:00:46 +00:00
p2p/tracker: fix crash in clean when tracker is stopped
This commit is contained in:
parent
723aae2b4e
commit
1a482cac6a
2 changed files with 23 additions and 0 deletions
|
|
@ -138,6 +138,10 @@ func (t *Tracker) clean() {
|
||||||
t.lock.Lock()
|
t.lock.Lock()
|
||||||
defer t.lock.Unlock()
|
defer t.lock.Unlock()
|
||||||
|
|
||||||
|
if t.expire == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// Expire anything within a certain threshold (might be no items at all if
|
// Expire anything within a certain threshold (might be no items at all if
|
||||||
// we raced with the delivery)
|
// we raced with the delivery)
|
||||||
for t.expire.Len() > 0 {
|
for t.expire.Len() > 0 {
|
||||||
|
|
|
||||||
|
|
@ -24,6 +24,25 @@ import (
|
||||||
"github.com/ethereum/go-ethereum/p2p"
|
"github.com/ethereum/go-ethereum/p2p"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// TestCleanAfterStop verifies that the clean method does not crash when called
|
||||||
|
// after Stop. This can happen because clean is scheduled via time.AfterFunc and
|
||||||
|
// may fire after Stop sets t.expire to nil.
|
||||||
|
func TestCleanAfterStop(t *testing.T) {
|
||||||
|
cap := p2p.Cap{Name: "test", Version: 1}
|
||||||
|
timeout := 50 * time.Millisecond
|
||||||
|
tr := New(cap, "peer1", timeout)
|
||||||
|
|
||||||
|
// Track a request to start the expiration timer.
|
||||||
|
tr.Track(Request{ID: 1, ReqCode: 0x01, RespCode: 0x02, Size: 1})
|
||||||
|
|
||||||
|
// Stop the tracker, then wait for the timer to fire.
|
||||||
|
tr.Stop()
|
||||||
|
time.Sleep(timeout + 50*time.Millisecond)
|
||||||
|
|
||||||
|
// Also verify that calling clean directly after stop doesn't panic.
|
||||||
|
tr.clean()
|
||||||
|
}
|
||||||
|
|
||||||
// This checks that metrics gauges for pending requests are be decremented when a
|
// This checks that metrics gauges for pending requests are be decremented when a
|
||||||
// Tracker is stopped.
|
// Tracker is stopped.
|
||||||
func TestMetricsOnStop(t *testing.T) {
|
func TestMetricsOnStop(t *testing.T) {
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue