mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-29 13:05:46 +00:00
ipn/ipnserver: move the unserved connection logic to a Listener
So future refactors can only deal with a net.Listener and be unconcerned with their caller's (Windows-specific) struggles. Change-Id: I0af588b9a769ab65c59b0bd21f8a0c99abfa1784 Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
This commit is contained in:
parent
c23a378f63
commit
d381bc2b6c
@ -718,6 +718,12 @@ func Run(ctx context.Context, logf logger.Logf, logid string, getEngine func() (
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if unservedConn != nil {
|
||||||
|
listen = &listenerWithReadyConn{
|
||||||
|
Listener: listen,
|
||||||
|
c: unservedConn,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
b, err := ipnlocal.NewLocalBackend(logf, logid, store, eng)
|
b, err := ipnlocal.NewLocalBackend(logf, logid, store, eng)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -758,14 +764,7 @@ func Run(ctx context.Context, logf logger.Logf, logid string, getEngine func() (
|
|||||||
|
|
||||||
systemd.Ready()
|
systemd.Ready()
|
||||||
for i := 1; ctx.Err() == nil; i++ {
|
for i := 1; ctx.Err() == nil; i++ {
|
||||||
var c net.Conn
|
c, err := listen.Accept()
|
||||||
var err error
|
|
||||||
if unservedConn != nil {
|
|
||||||
c = unservedConn
|
|
||||||
unservedConn = nil
|
|
||||||
} else {
|
|
||||||
c, err = listen.Accept()
|
|
||||||
}
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if ctx.Err() == nil {
|
if ctx.Err() == nil {
|
||||||
logf("ipnserver: Accept: %v", err)
|
logf("ipnserver: Accept: %v", err)
|
||||||
@ -1031,3 +1030,23 @@ func marshalNotify(n ipn.Notify, logf logger.Logf) (b []byte, ok bool) {
|
|||||||
}
|
}
|
||||||
return b, true
|
return b, true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// listenerWithReadyConn is a net.Listener wrapper that has
|
||||||
|
// one net.Conn ready to be accepted already.
|
||||||
|
type listenerWithReadyConn struct {
|
||||||
|
net.Listener
|
||||||
|
|
||||||
|
mu sync.Mutex
|
||||||
|
c net.Conn // if non-nil, ready to be Accepted
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ln *listenerWithReadyConn) Accept() (net.Conn, error) {
|
||||||
|
ln.mu.Lock()
|
||||||
|
c := ln.c
|
||||||
|
ln.c = nil
|
||||||
|
ln.mu.Unlock()
|
||||||
|
if c != nil {
|
||||||
|
return c, nil
|
||||||
|
}
|
||||||
|
return ln.Listener.Accept()
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user