github.com/metacubex/sing-shadowsocks2@v0.2.0/shadowaead/method_wait.go (about)

     1  package shadowaead
     2  
     3  import (
     4  	"github.com/sagernet/sing/common/buf"
     5  	"github.com/sagernet/sing/common/bufio"
     6  	M "github.com/sagernet/sing/common/metadata"
     7  	N "github.com/sagernet/sing/common/network"
     8  )
     9  
    10  var _ N.ReadWaiter = (*clientConn)(nil)
    11  
    12  func (c *clientConn) InitializeReadWaiter(options N.ReadWaitOptions) (needCopy bool) {
    13  	if c.reader == nil {
    14  		c.readWaitOptions = options
    15  		return options.NeedHeadroom()
    16  	}
    17  	return c.reader.InitializeReadWaiter(options)
    18  }
    19  
    20  func (c *clientConn) WaitReadBuffer() (buffer *buf.Buffer, err error) {
    21  	if c.reader == nil {
    22  		err = c.readResponse()
    23  		if err != nil {
    24  			return
    25  		}
    26  	}
    27  	return c.reader.WaitReadBuffer()
    28  }
    29  
    30  var _ N.PacketReadWaitCreator = (*clientPacketConn)(nil)
    31  
    32  func (c *clientPacketConn) CreateReadWaiter() (N.PacketReadWaiter, bool) {
    33  	readWaiter, isReadWaiter := bufio.CreateReadWaiter(c.reader)
    34  	if !isReadWaiter {
    35  		return nil, false
    36  	}
    37  	return &clientPacketReadWaiter{c, readWaiter}, true
    38  }
    39  
    40  var _ N.PacketReadWaiter = (*clientPacketReadWaiter)(nil)
    41  
    42  type clientPacketReadWaiter struct {
    43  	*clientPacketConn
    44  	readWaiter N.ReadWaiter
    45  }
    46  
    47  func (w *clientPacketReadWaiter) InitializeReadWaiter(options N.ReadWaitOptions) (needCopy bool) {
    48  	return w.readWaiter.InitializeReadWaiter(options)
    49  }
    50  
    51  func (w *clientPacketReadWaiter) WaitReadPacket() (buffer *buf.Buffer, destination M.Socksaddr, err error) {
    52  	buffer, err = w.readWaiter.WaitReadBuffer()
    53  	if err != nil {
    54  		return
    55  	}
    56  	destination, err = w.readPacket(buffer)
    57  	if err != nil {
    58  		buffer.Release()
    59  		return nil, M.Socksaddr{}, err
    60  	}
    61  	return
    62  }