sing-box/transport/v2raygrpclite/conn.go

172 lines
3.7 KiB
Go
Raw Normal View History

2022-08-27 13:05:15 +00:00
package v2raygrpclite
import (
std_bufio "bufio"
"bytes"
"encoding/binary"
"io"
"net"
"net/http"
"os"
2022-09-13 02:41:10 +00:00
"sync"
2022-08-27 13:05:15 +00:00
"time"
2022-08-29 02:10:41 +00:00
"github.com/sagernet/sing-box/common/baderror"
2022-08-27 13:05:15 +00:00
"github.com/sagernet/sing/common"
"github.com/sagernet/sing/common/buf"
"github.com/sagernet/sing/common/bufio"
"github.com/sagernet/sing/common/rw"
)
// kanged from: https://github.com/Qv2ray/gun-lite
var _ net.Conn = (*GunConn)(nil)
type GunConn struct {
reader *std_bufio.Reader
writer io.Writer
flusher http.Flusher
create chan struct{}
err error
readRemaining int
2022-09-13 02:41:10 +00:00
writeAccess sync.Mutex
2022-08-27 13:05:15 +00:00
}
func newGunConn(reader io.Reader, writer io.Writer, flusher http.Flusher) *GunConn {
return &GunConn{
reader: std_bufio.NewReader(reader),
writer: writer,
flusher: flusher,
}
}
func newLateGunConn(writer io.Writer) *GunConn {
return &GunConn{
create: make(chan struct{}),
writer: writer,
}
}
func (c *GunConn) setup(reader io.Reader, err error) {
c.reader = std_bufio.NewReader(reader)
c.err = err
close(c.create)
}
func (c *GunConn) Read(b []byte) (n int, err error) {
n, err = c.read(b)
2022-08-29 02:10:41 +00:00
return n, baderror.WrapH2(err)
2022-08-27 13:05:15 +00:00
}
func (c *GunConn) read(b []byte) (n int, err error) {
if c.reader == nil {
<-c.create
if c.err != nil {
return 0, c.err
}
}
if c.readRemaining > 0 {
if len(b) > c.readRemaining {
b = b[:c.readRemaining]
}
n, err = c.reader.Read(b)
c.readRemaining -= n
return
}
_, err = c.reader.Discard(6)
if err != nil {
return
}
dataLen, err := binary.ReadUvarint(c.reader)
if err != nil {
return
}
readLen := int(dataLen)
c.readRemaining = readLen
if len(b) > readLen {
b = b[:readLen]
}
n, err = c.reader.Read(b)
c.readRemaining -= n
return
}
func (c *GunConn) Write(b []byte) (n int, err error) {
protobufHeader := [1 + binary.MaxVarintLen64]byte{0x0A}
varuintLen := binary.PutUvarint(protobufHeader[1:], uint64(len(b)))
grpcHeader := buf.Get(5)
grpcPayloadLen := uint32(1 + varuintLen + len(b))
binary.BigEndian.PutUint32(grpcHeader[1:5], grpcPayloadLen)
2022-09-13 02:41:10 +00:00
c.writeAccess.Lock()
2022-08-27 13:05:15 +00:00
_, err = bufio.Copy(c.writer, io.MultiReader(bytes.NewReader(grpcHeader), bytes.NewReader(protobufHeader[:varuintLen+1]), bytes.NewReader(b)))
2022-09-13 02:41:10 +00:00
c.writeAccess.Unlock()
2022-08-27 13:05:15 +00:00
buf.Put(grpcHeader)
2022-11-23 06:56:31 +00:00
if err == nil && c.flusher != nil {
c.flusher.Flush()
2022-08-27 13:05:15 +00:00
}
2022-08-29 02:10:41 +00:00
return len(b), baderror.WrapH2(err)
2022-08-27 13:05:15 +00:00
}
func (c *GunConn) WriteBuffer(buffer *buf.Buffer) error {
defer buffer.Release()
dataLen := buffer.Len()
2022-11-23 06:56:31 +00:00
varLen := rw.UVariantLen(uint64(dataLen))
2022-08-27 13:05:15 +00:00
header := buffer.ExtendHeader(6 + varLen)
binary.BigEndian.PutUint32(header[1:5], uint32(1+varLen+dataLen))
header[5] = 0x0A
binary.PutUvarint(header[6:], uint64(dataLen))
err := rw.WriteBytes(c.writer, buffer.Bytes())
2022-11-23 06:56:31 +00:00
if err == nil && c.flusher != nil {
2022-08-27 13:05:15 +00:00
c.flusher.Flush()
}
2022-08-29 02:10:41 +00:00
return baderror.WrapH2(err)
2022-08-27 13:05:15 +00:00
}
func (c *GunConn) FrontHeadroom() int {
return 6 + binary.MaxVarintLen64
}
func (c *GunConn) Close() error {
return common.Close(c.reader, c.writer)
}
func (c *GunConn) LocalAddr() net.Addr {
return nil
}
func (c *GunConn) RemoteAddr() net.Addr {
return nil
}
func (c *GunConn) SetDeadline(t time.Time) error {
2022-11-23 06:56:31 +00:00
if responseWriter, loaded := c.writer.(interface {
2022-11-11 14:08:20 +00:00
SetWriteDeadline(time.Time) error
2022-11-23 06:56:31 +00:00
}); loaded {
return responseWriter.SetWriteDeadline(t)
2022-11-11 14:08:20 +00:00
}
2022-11-23 06:56:31 +00:00
return os.ErrInvalid
2022-08-27 13:05:15 +00:00
}
func (c *GunConn) SetReadDeadline(t time.Time) error {
2022-11-23 06:56:31 +00:00
if responseWriter, loaded := c.writer.(interface {
2022-11-11 14:08:20 +00:00
SetReadDeadline(time.Time) error
2022-11-23 06:56:31 +00:00
}); loaded {
return responseWriter.SetReadDeadline(t)
2022-11-11 14:08:20 +00:00
}
2022-11-23 06:56:31 +00:00
return os.ErrInvalid
2022-08-27 13:05:15 +00:00
}
func (c *GunConn) SetWriteDeadline(t time.Time) error {
2022-11-23 06:56:31 +00:00
if responseWriter, loaded := c.writer.(interface {
2022-11-11 14:08:20 +00:00
SetWriteDeadline(time.Time) error
2022-11-23 06:56:31 +00:00
}); loaded {
return responseWriter.SetWriteDeadline(t)
2022-11-11 14:08:20 +00:00
}
2022-11-23 06:56:31 +00:00
return os.ErrInvalid
2022-08-27 13:05:15 +00:00
}