Unverified Commit 8f11d279 authored by Evolution404's avatar Evolution404 Committed by GitHub

p2p/simulations: fix unlikely crash in probabilistic connect (#23200)

When the nodeCount is less than 10, it will panic with the out of bound error.
How about we just skip this round, when rand1 and rand2 are equal?
parent b157bae2
...@@ -123,20 +123,12 @@ func probabilistic(net *Network, quit chan struct{}, nodeCount int) { ...@@ -123,20 +123,12 @@ func probabilistic(net *Network, quit chan struct{}, nodeCount int) {
randWait := time.Duration(rand.Intn(5000)+1000) * time.Millisecond randWait := time.Duration(rand.Intn(5000)+1000) * time.Millisecond
rand1 := rand.Intn(nodeCount - 1) rand1 := rand.Intn(nodeCount - 1)
rand2 := rand.Intn(nodeCount - 1) rand2 := rand.Intn(nodeCount - 1)
if rand1 < rand2 { if rand1 <= rand2 {
lowid = rand1 lowid = rand1
highid = rand2 highid = rand2
} else if rand1 > rand2 { } else if rand1 > rand2 {
highid = rand1 highid = rand1
lowid = rand2 lowid = rand2
} else {
if rand1 == 0 {
rand2 = 9
} else if rand1 == 9 {
rand1 = 0
}
lowid = rand1
highid = rand2
} }
var steps = highid - lowid var steps = highid - lowid
wg.Add(steps) wg.Add(steps)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment