wgengine/monitor: make Darwin monitor shut down cleanly, add test
Don't use os.NewFile or (*os.File).Close on the AF_ROUTE socket. It apparently does weird things to the fd and at least doesn't seem to close it. Just use the unix package. The test doesn't actually fail reliably before the fix, though. It was an attempt. But this fixes the integration tests. Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>pull/1422/head
parent
625c413508
commit
24fa616e73
|
@ -7,7 +7,7 @@ package monitor
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"sync"
|
||||||
|
|
||||||
"golang.org/x/net/route"
|
"golang.org/x/net/route"
|
||||||
"golang.org/x/sys/unix"
|
"golang.org/x/sys/unix"
|
||||||
|
@ -31,22 +31,27 @@ func newOSMon(logf logger.Logf) (osMon, error) {
|
||||||
}
|
}
|
||||||
return &darwinRouteMon{
|
return &darwinRouteMon{
|
||||||
logf: logf,
|
logf: logf,
|
||||||
f: os.NewFile(uintptr(fd), "AF_ROUTE"),
|
fd: fd,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type darwinRouteMon struct {
|
type darwinRouteMon struct {
|
||||||
logf logger.Logf
|
logf logger.Logf
|
||||||
f *os.File // AF_ROUTE socket
|
fd int // AF_ROUTE socket
|
||||||
buf [2 << 10]byte
|
buf [2 << 10]byte
|
||||||
|
closeOnce sync.Once
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *darwinRouteMon) Close() error {
|
func (m *darwinRouteMon) Close() error {
|
||||||
return m.f.Close()
|
var err error
|
||||||
|
m.closeOnce.Do(func() {
|
||||||
|
err = unix.Close(m.fd)
|
||||||
|
})
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *darwinRouteMon) Receive() (message, error) {
|
func (m *darwinRouteMon) Receive() (message, error) {
|
||||||
n, err := m.f.Read(m.buf[:])
|
n, err := unix.Read(m.fd, m.buf[:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,30 @@
|
||||||
|
// Copyright (c) 2021 Tailscale Inc & AUTHORS All rights reserved.
|
||||||
|
// Use of this source code is governed by a BSD-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package monitor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMonitorStartClose(t *testing.T) {
|
||||||
|
mon, err := New(t.Logf)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
mon.Start()
|
||||||
|
if err := mon.Close(); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMonitorJustClose(t *testing.T) {
|
||||||
|
mon, err := New(t.Logf)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if err := mon.Close(); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue