control/controlclient: run SetControlClientStatus in goroutine

We have cases where the SetControlClientStatus would result in
a Shutdown call back into the auto client that would block
forever. The right thing to do here is to fix the LocalBackend
state machine but thats a different dumpster fire that we
are slowly making progress towards.

This makes it so that the SetControlClientStatus happens in a
different goroutine so that calls back into the auto client
do not block.

Also add a few missing mu.Unlocks in LocalBackend.Start.

Updates #9181

Signed-off-by: Maisem Ali <maisem@tailscale.com>
pull/9190/head
Maisem Ali 9 months ago committed by Maisem Ali
parent 86dc0af5ae
commit 7074a40c06

@ -159,7 +159,6 @@ type Auto struct {
loggedIn bool // true if currently logged in loggedIn bool // true if currently logged in
loginGoal *LoginGoal // non-nil if some login activity is desired loginGoal *LoginGoal // non-nil if some login activity is desired
synced bool // true if our netmap is up-to-date synced bool // true if our netmap is up-to-date
inSendStatus int // number of sendStatus calls currently in progress
state State state State
authCtx context.Context // context used for auth requests authCtx context.Context // context used for auth requests
@ -646,7 +645,6 @@ func (c *Auto) sendStatus(who string, err error, url string, nm *netmap.NetworkM
state := c.state state := c.state
loggedIn := c.loggedIn loggedIn := c.loggedIn
synced := c.synced synced := c.synced
c.inSendStatus++
c.mu.Unlock() c.mu.Unlock()
c.logf("[v1] sendStatus: %s: %v", who, state) c.logf("[v1] sendStatus: %s: %v", who, state)
@ -666,11 +664,10 @@ func (c *Auto) sendStatus(who string, err error, url string, nm *netmap.NetworkM
Err: err, Err: err,
state: state, state: state,
} }
c.observer.SetControlClientStatus(new)
c.mu.Lock() // Launch a new goroutine to avoid blocking the caller while the observer
c.inSendStatus-- // does its thing, which may result in a call back into the client.
c.mu.Unlock() go c.observer.SetControlClientStatus(new)
} }
func (c *Auto) Login(t *tailcfg.Oauth2Token, flags LoginFlags) { func (c *Auto) Login(t *tailcfg.Oauth2Token, flags LoginFlags) {
@ -732,7 +729,6 @@ func (c *Auto) Shutdown() {
c.logf("client.Shutdown()") c.logf("client.Shutdown()")
c.mu.Lock() c.mu.Lock()
inSendStatus := c.inSendStatus
closed := c.closed closed := c.closed
direct := c.direct direct := c.direct
if !closed { if !closed {
@ -742,7 +738,7 @@ func (c *Auto) Shutdown() {
} }
c.mu.Unlock() c.mu.Unlock()
c.logf("client.Shutdown: inSendStatus=%v", inSendStatus) c.logf("client.Shutdown")
if !closed { if !closed {
c.unregisterHealthWatch() c.unregisterHealthWatch()
close(c.quit) close(c.quit)

@ -1396,6 +1396,7 @@ func (b *LocalBackend) Start(opts ipn.Options) error {
wantRunning := prefs.WantRunning() wantRunning := prefs.WantRunning()
if wantRunning { if wantRunning {
if err := b.initMachineKeyLocked(); err != nil { if err := b.initMachineKeyLocked(); err != nil {
b.mu.Unlock()
return fmt.Errorf("initMachineKeyLocked: %w", err) return fmt.Errorf("initMachineKeyLocked: %w", err)
} }
} }

Loading…
Cancel
Save