世界
8c5996c6bc
The plain OP_REQ_IMPORT + TryReserveForImport path already delivers race-free exclusive access for every observable feature; the lease two-phase commit (OP_REQ_IMPORT_EXT, control lease frames, serverImportLease) only added identity-replacement detection and pre-import busy advertisement, neither of which is asserted by any feature test. TryReserveForImport now runs as a single critical section under inventoryAccess, closing the TOCTOU window the LeaseIdentity recheck used to guard. Export interface loses LeaseIdentity/LeaseCheck. controlPingLoop writes directly via writeControlMessage, dropping the clientControlSession holder and the controlSession/controlAccess fields on ClientService. Control protocol drops the ImportLease capability bit; controlExtensionCapabilities is now PayloadFrames | DeviceStateV2.
78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
//go:build linux || (darwin && cgo) || windows
|
|
|
|
package usbip
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
E "github.com/sagernet/sing/common/exceptions"
|
|
)
|
|
|
|
func (s *ServerService) acceptLoop(ln net.Listener) {
|
|
for {
|
|
conn, err := ln.Accept()
|
|
if err != nil {
|
|
select {
|
|
case <-s.ctx.Done():
|
|
return
|
|
default:
|
|
}
|
|
if E.IsClosed(err) {
|
|
return
|
|
}
|
|
//nolint:staticcheck // net.Error.Temporary predates net.ErrClosed; replacement needs a separate audit.
|
|
if netError, isNetError := err.(net.Error); isNetError && netError.Temporary() {
|
|
s.logger.Error("accept: ", err)
|
|
if !sleepCtx(s.ctx, 200*time.Millisecond) {
|
|
return
|
|
}
|
|
continue
|
|
}
|
|
s.logger.Error("accept: ", err)
|
|
return
|
|
}
|
|
go s.dispatchConn(conn)
|
|
}
|
|
}
|
|
|
|
func (s *ServerService) dispatchConn(conn net.Conn) {
|
|
cancelClose := closeConnOnContextDone(s.ctx, conn)
|
|
defer cancelClose()
|
|
var prefix [controlPrefaceSize]byte
|
|
_, err := io.ReadFull(conn, prefix[:])
|
|
if err != nil {
|
|
s.logger.Debug("read connection preface: ", err)
|
|
_ = conn.Close()
|
|
return
|
|
}
|
|
if bytes.Equal(prefix[:], controlPreface[:]) {
|
|
s.handleControlConn(conn)
|
|
return
|
|
}
|
|
s.handleStandardConn(conn, ParseOpHeader(prefix[:]))
|
|
}
|
|
|
|
func (s *ServerService) readControlConn(sub *exportSubscriber, done chan<- struct{}) {
|
|
defer close(done)
|
|
var reader controlReader
|
|
for {
|
|
message, err := reader.read(sub.conn)
|
|
if err != nil {
|
|
return
|
|
}
|
|
frame := message.Frame
|
|
switch frame.Type {
|
|
case controlFramePing:
|
|
s.ledger.enqueueFrame(sub, controlFrame{
|
|
Type: controlFramePong,
|
|
Version: controlProtocolVersion,
|
|
})
|
|
default:
|
|
return
|
|
}
|
|
}
|
|
}
|