chunyu/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2025-09-10 05:14:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve ktls rx error handling

Browse Source
This commit is contained in:
世界 2025-09-09 22:20:53 +08:00
parent 71fc261f91
commit c5a4c6bead
No known key found for this signature in database
GPG Key ID: CD109927C34A63C4
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/ktls/ktls.go
View File

@ -32,6 +32,7 @@ type Conn struct {
readWaitOptions N.ReadWaitOptions
kernelTx bool
kernelRx bool
pendingRxSplice bool
}
func NewConn(ctx context.Context, logger logger.ContextLogger, conn aTLS.Conn, txOffload, rxOffload bool) (aTLS.Conn, error) {
@ -103,6 +104,7 @@ func (c *Conn) SyscallConnForRead() syscall.RawConn {
func (c *Conn) HandleSyscallReadError(inputErr error) ([]byte, error) {
if errors.Is(inputErr, unix.EINVAL) {
c.pendingRxSplice = true
err := c.readRecord()
if err != nil {
return nil, E.Cause(err, "ktls: handle non-application-data record")

6
common/ktls/ktls_linux.go
View File

@ -258,14 +258,14 @@ func (c *Conn) readKernelRecord() (uint8, []byte, error) {
var err error
er := c.rawSyscallConn.Read(func(fd uintptr) bool {
n, err = recvmsg(int(fd), &msg, 0)
return err != unix.EAGAIN
return err != unix.EAGAIN || c.pendingRxSplice
})
if er != nil {
return 0, nil, er
}
switch err {
case nil:
case syscall.EINVAL:
case syscall.EINVAL, syscall.EAGAIN:
return 0, nil, c.rawConn.In.SetErrorLocked(c.sendAlert(alertProtocolVersion))
case syscall.EMSGSIZE:
return 0, nil, c.rawConn.In.SetErrorLocked(c.sendAlert(alertRecordOverflow))
@ -276,7 +276,7 @@ func (c *Conn) readKernelRecord() (uint8, []byte, error) {
}
if n <= 0 {
return 0, nil, io.EOF
return 0, nil, c.rawConn.In.SetErrorLocked(io.EOF)
}
if cmsg.Level == unix.SOL_TLS && cmsg.Type == TLS_GET_RECORD_TYPE {