mirror of
https://github.com/netbirdio/netbird.git
synced 2026-09-23 07:09:08 +02:00
46 lines
1.7 KiB
Go
46 lines
1.7 KiB
Go
package server
|
|
|
|
import "context"
|
|
|
|
// sessionAgent abstracts the per-platform manager that spawns and tracks
|
|
// the user-session VNC agent. Resolve returns the agent's Unix-socket
|
|
// path, the shared per-spawn token, and the uid the agent was spawned
|
|
// under (used to validate peer credentials before the daemon hands the
|
|
// token to whoever is on the other end of the socket). Resolve may spawn
|
|
// the agent lazily.
|
|
// Release reports that one proxied connection is done with the agent, so a
|
|
// platform that recycles the agent per connection can tear it down once the last
|
|
// one is gone. Every successful Resolve owes exactly one Release.
|
|
type sessionAgent interface {
|
|
Resolve(ctx context.Context) (socketPath, token string, peerUID uint32, err error)
|
|
Release()
|
|
}
|
|
|
|
// resetServiceAgent reopens the latch stopServiceAgent closed, so a restarted
|
|
// server can build a manager again. Without it every accept loop of the new
|
|
// lifecycle keeps getting a nil agent and service mode stays dead for the rest
|
|
// of the process. Owned by Start, mirroring the other stop-time latches it
|
|
// clears.
|
|
func (s *Server) resetServiceAgent() {
|
|
s.serviceAgentMu.Lock()
|
|
defer s.serviceAgentMu.Unlock()
|
|
s.serviceAgentStopped = false
|
|
}
|
|
|
|
// stopServiceAgent tears down the shared manager, if one was ever built, and
|
|
// latches the server so a still-draining accept loop cannot build another.
|
|
// Owned by Stop rather than by an accept loop: the loops share the manager, so
|
|
// the first one to exit must not take it away from the others.
|
|
func (s *Server) stopServiceAgent() {
|
|
s.serviceAgentMu.Lock()
|
|
stop := s.serviceAgentStop
|
|
s.serviceAgentStop = nil
|
|
s.serviceAgentMgr = nil
|
|
s.serviceAgentStopped = true
|
|
s.serviceAgentMu.Unlock()
|
|
|
|
if stop != nil {
|
|
stop()
|
|
}
|
|
}
|