bff2ba70fd
This commit introduces package connmgr which contains connection management related functionality. The following is an overview of the features the package provides: - Maintain fixed number of outbound connections - Optional connect-only mode - Retry persistent connections with increasing back-off - Source peers from DNS seeds - Use Tor to resolve DNS - Dynamic ban scores - Test coverage In addition, btcd has been refactored to make use of the new package by extending the connection manager to work with the server to source and maintain peer connections. The following is a broad overview of the changes to integrate the package: - Simplify peer state by removing pending, retry peers - Refactor to remove retries which are now handled by connmgr - Use callback to add addresses sourced from the DNS seed Finally the following connection-related things have been improved as a part of this refactor: - Fixes 100% cpu usage when network is down (#129) - Fixes issues with max peers (#577) - Simplify outbound peer connections management
68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
// Copyright (c) 2016 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package connmgr
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// TestDynamicBanScoreDecay tests the exponential decay implemented in
|
|
// DynamicBanScore.
|
|
func TestDynamicBanScoreDecay(t *testing.T) {
|
|
var bs DynamicBanScore
|
|
base := time.Now()
|
|
|
|
r := bs.increase(100, 50, base)
|
|
if r != 150 {
|
|
t.Errorf("Unexpected result %d after ban score increase.", r)
|
|
}
|
|
|
|
r = bs.int(base.Add(time.Minute))
|
|
if r != 125 {
|
|
t.Errorf("Halflife check failed - %d instead of 125", r)
|
|
}
|
|
|
|
r = bs.int(base.Add(7 * time.Minute))
|
|
if r != 100 {
|
|
t.Errorf("Decay after 7m - %d instead of 100", r)
|
|
}
|
|
}
|
|
|
|
// TestDynamicBanScoreLifetime tests that DynamicBanScore properly yields zero
|
|
// once the maximum age is reached.
|
|
func TestDynamicBanScoreLifetime(t *testing.T) {
|
|
var bs DynamicBanScore
|
|
base := time.Now()
|
|
|
|
r := bs.increase(0, math.MaxUint32, base)
|
|
r = bs.int(base.Add(Lifetime * time.Second))
|
|
if r != 3 { // 3, not 4 due to precision loss and truncating 3.999...
|
|
t.Errorf("Pre max age check with MaxUint32 failed - %d", r)
|
|
}
|
|
r = bs.int(base.Add((Lifetime + 1) * time.Second))
|
|
if r != 0 {
|
|
t.Errorf("Zero after max age check failed - %d instead of 0", r)
|
|
}
|
|
}
|
|
|
|
// TestDynamicBanScore tests exported functions of DynamicBanScore. Exponential
|
|
// decay or other time based behavior is tested by other functions.
|
|
func TestDynamicBanScoreReset(t *testing.T) {
|
|
var bs DynamicBanScore
|
|
if bs.Int() != 0 {
|
|
t.Errorf("Initial state is not zero.")
|
|
}
|
|
bs.Increase(100, 0)
|
|
r := bs.Int()
|
|
if r != 100 {
|
|
t.Errorf("Unexpected result %d after ban score increase.", r)
|
|
}
|
|
bs.Reset()
|
|
if bs.Int() != 0 {
|
|
t.Errorf("Failed to reset ban score.")
|
|
}
|
|
}
|