mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-12 17:59:06 +02:00
Zoltán Papp
5df35e3e27
Nothing recorded which run of the connection was current, so three defects followed from the same gap. A ConnectClient is single-use, and the daemon builds a fresh one per outer-retry turn (server.go connect). Each turn overwrote s.connectClient and nothing stopped the one it replaced — the outgoing run loop had returned, which is what brought control back to the retry, but that was assumed rather than enforced, and any teardown its error path left half-done got no second chance. cleanupConnection read s.connectClient, cancelled, then stopped that engine. Nothing established the client it read was still current by the time it stopped it, so a teardown could target a client a newer turn had already replaced and leave the live one running untracked. Down's wait on clientGiveUpChan and Up's refusal to start a second loop kept the window narrow, but by arrangement rather than by construction. Third, the engine was stopped twice concurrently: actCancel woke the run loop, which stops the engine on its way out, while cleanupConnection stopped the same engine directly. The TODO there said ConnectClient.Stop was the right call and that its unbounded wait was what ruled it out. RunSupervisor records the generation of the current run. Publish refuses a client from a superseded run and stops the client it displaces, so no ConnectClient is dropped without being stopped. Stop invalidates whatever run is in flight, stops the published client and waits for the run to exit. ConnectClient.StopWithContext bounds that wait, which removes the TODO's obstacle: cleanupConnection now hands the run loop sole ownership of engine shutdown and passes Down's 5s budget down. Stop() keeps its signature and its unbounded wait, so callers outside this change are untouched. embed.Client.Stop had built the same bound by hand with a goroutine and a select purely to watch its caller's context; it passes the context down instead. clientGiveUpChan and connectClient are gone — the supervisor answers both. The MDM restart path drops its hand-rolled 10s channel wait for the same Stop, which additionally stops the client the previous run left behind. Its deliberate choice to leave clientRunning set is unchanged. Down now waits inside cleanupConnection, under s.mutex, where it previously waited after releasing it. That is what pins the client being stopped to the one current when the call started; the cost is that Down can hold the mutex for up to its 5s budget. Found while fixing the iOS wifi-to-cellular black-hole (#7329), which was the same class of defect in the mobile SDKs. No bug report backs the daemon findings — they are read off the code, and the narrow windows above may be why they have not been observed.
117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newSupervisedClient() *ConnectClient {
|
|
return NewConnectClient(context.Background(), nil, nil)
|
|
}
|
|
|
|
func TestRunSupervisorPublishRejectsSupersededRun(t *testing.T) {
|
|
var s RunSupervisor
|
|
ctx := context.Background()
|
|
|
|
staleGen, staleDone := s.Begin()
|
|
defer close(staleDone)
|
|
|
|
freshGen, freshDone := s.Begin()
|
|
defer close(freshDone)
|
|
|
|
fresh := newSupervisedClient()
|
|
require.True(t, s.Publish(ctx, freshGen, fresh))
|
|
|
|
assert.False(t, s.Publish(ctx, staleGen, newSupervisedClient()))
|
|
assert.Same(t, fresh, s.Current())
|
|
}
|
|
|
|
func TestRunSupervisorPublishStopsDisplacedClient(t *testing.T) {
|
|
var s RunSupervisor
|
|
ctx := context.Background()
|
|
|
|
generation, done := s.Begin()
|
|
defer close(done)
|
|
|
|
displaced := newSupervisedClient()
|
|
require.True(t, s.Publish(ctx, generation, displaced))
|
|
|
|
replacement := newSupervisedClient()
|
|
require.True(t, s.Publish(ctx, generation, replacement))
|
|
|
|
assert.Same(t, replacement, s.Current())
|
|
assert.Error(t, displaced.ctx.Err(), "the displaced client's run context should be cancelled")
|
|
}
|
|
|
|
func TestRunSupervisorStopInvalidatesRunInFlight(t *testing.T) {
|
|
var s RunSupervisor
|
|
ctx := context.Background()
|
|
|
|
generation, done := s.Begin()
|
|
close(done)
|
|
|
|
require.NoError(t, s.Stop(ctx))
|
|
|
|
assert.False(t, s.Publish(ctx, generation, newSupervisedClient()))
|
|
assert.Nil(t, s.Current())
|
|
}
|
|
|
|
func TestRunSupervisorStopWaitsForRunExit(t *testing.T) {
|
|
var s RunSupervisor
|
|
|
|
_, done := s.Begin()
|
|
|
|
stopped := make(chan error, 1)
|
|
go func() { stopped <- s.Stop(context.Background()) }()
|
|
|
|
select {
|
|
case <-stopped:
|
|
t.Fatal("Stop returned before the run signalled its exit")
|
|
case <-time.After(50 * time.Millisecond):
|
|
}
|
|
|
|
close(done)
|
|
|
|
select {
|
|
case err := <-stopped:
|
|
require.NoError(t, err)
|
|
case <-time.After(time.Second):
|
|
t.Fatal("Stop did not return after the run exited")
|
|
}
|
|
}
|
|
|
|
func TestRunSupervisorStopGivesUpWaitOnContext(t *testing.T) {
|
|
var s RunSupervisor
|
|
|
|
_, done := s.Begin()
|
|
defer close(done)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
|
defer cancel()
|
|
|
|
assert.ErrorIs(t, s.Stop(ctx), context.DeadlineExceeded)
|
|
}
|
|
|
|
func TestRunSupervisorAliveTracksRun(t *testing.T) {
|
|
var s RunSupervisor
|
|
|
|
assert.False(t, s.Alive(), "no run has started")
|
|
|
|
_, done := s.Begin()
|
|
assert.True(t, s.Alive(), "a run is in flight")
|
|
|
|
close(done)
|
|
assert.False(t, s.Alive(), "the run signalled its exit")
|
|
}
|
|
|
|
func TestRunSupervisorStopWithoutRunIsNoop(t *testing.T) {
|
|
var s RunSupervisor
|
|
|
|
require.NoError(t, s.Stop(context.Background()))
|
|
assert.Nil(t, s.Current())
|
|
}
|