世界
cec39eb00c
Collapse ServerService and ClientService to one definition each by hiding Linux vs Darwin behind ExportHost, ImportHost, and Export seams. Along the way, extract three state machines that previously lived as scattered fields on the service structs: - LeaseManager owns its own mutex and closes the availability-vs-insert TOCTOU by checking export busy inside Issue under the same lock as the insert. - DataSession gives the three per-import data-plane implementations (Linux kernel handoff, Darwin server data session, Darwin virtual controller) a uniform Done/Err/Close interface. - clientAssignment encapsulates the matched/import-all target state and exposes ApplyMatched/ApplyAll diffs to ClientService, which keeps worker goroutine lifecycle. Service busy tracking moves off the per-platform serverExport struct onto ServerService.busy, since it follows the lease/import lifecycle rather than physical claim/release. linux_test.go is migrated to construct ServerService and ClientService through the new host interfaces.
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
//go:build linux
|
|
|
|
package usbip
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
const ueventReceiveBufferSize = 1 << 20
|
|
|
|
type ueventListener struct {
|
|
fd int
|
|
}
|
|
|
|
func newUEventListener() (*ueventListener, error) {
|
|
fd, err := unix.Socket(unix.AF_NETLINK, unix.SOCK_DGRAM, unix.NETLINK_KOBJECT_UEVENT)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_ = unix.SetsockoptInt(fd, unix.SOL_SOCKET, unix.SO_RCVBUF, ueventReceiveBufferSize)
|
|
addr := &unix.SockaddrNetlink{
|
|
Family: unix.AF_NETLINK,
|
|
Groups: 1,
|
|
}
|
|
err = unix.Bind(fd, addr)
|
|
if err != nil {
|
|
_ = unix.Close(fd)
|
|
return nil, err
|
|
}
|
|
return &ueventListener{fd: fd}, nil
|
|
}
|
|
|
|
func (l *ueventListener) Close() error {
|
|
return unix.Close(l.fd)
|
|
}
|
|
|
|
func (l *ueventListener) WaitUSBEvent() error {
|
|
var buf [16384]byte
|
|
for {
|
|
n, from, err := unix.Recvfrom(l.fd, buf[:], 0)
|
|
if err == unix.ENOBUFS {
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if source, ok := from.(*unix.SockaddrNetlink); ok && source.Pid != 0 {
|
|
continue
|
|
}
|
|
if isUSBDeviceUEvent(buf[:n]) {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
var (
|
|
usbSubsystemMarker = []byte("\x00SUBSYSTEM=usb\x00")
|
|
usbDeviceTypeMarker = []byte("\x00DEVTYPE=usb_device\x00")
|
|
)
|
|
|
|
func isUSBDeviceUEvent(raw []byte) bool {
|
|
return bytes.Contains(raw, usbSubsystemMarker) && bytes.Contains(raw, usbDeviceTypeMarker)
|
|
}
|