SagerNet/sing-shadowtls
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-shadowtls.git synced 2025-04-05 05:07:36 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve v3 writer

Browse source
This commit is contained in:
世界 2023-05-31 10:57:48 +08:00
parent 2b3732dc6d
commit 74b210a0e6
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 31 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

28
v3_conn.go
View file

@ -17,7 +17,8 @@ import (
type verifiedConn struct { type verifiedConn struct {
net.Conn net.Conn
writer N.VectorisedWriter writer N.ExtendedWriter
vectorisedWriter N.VectorisedWriter
hmacAdd hash.Hash hmacAdd hash.Hash
hmacVerify hash.Hash hmacVerify hash.Hash
hmacIgnore hash.Hash hmacIgnore hash.Hash
@ -33,7 +34,8 @@ func newVerifiedConn(
) *verifiedConn { ) *verifiedConn {
return &verifiedConn{ return &verifiedConn{
Conn: conn, Conn: conn,
writer: bufio.NewVectorisedWriter(conn), writer: bufio.NewExtendedWriter(conn),
vectorisedWriter: bufio.NewVectorisedWriter(conn),
hmacAdd: hmacAdd, hmacAdd: hmacAdd,
hmacVerify: hmacVerify, hmacVerify: hmacVerify,
hmacIgnore: hmacIgnore, hmacIgnore: hmacIgnore,
@ -121,13 +123,27 @@ func (c *verifiedConn) write(p []byte) (n int, err error) {
hmacHash := c.hmacAdd.Sum(nil)[:hmacSize] hmacHash := c.hmacAdd.Sum(nil)[:hmacSize]
c.hmacAdd.Write(hmacHash) c.hmacAdd.Write(hmacHash)
copy(header[tlsHeaderSize:], hmacHash) copy(header[tlsHeaderSize:], hmacHash)
_, err = bufio.WriteVectorised(c.writer, [][]byte{common.Dup(header[:]), p}) _, err = bufio.WriteVectorised(c.vectorisedWriter, [][]byte{common.Dup(header[:]), p})
if err == nil { if err == nil {
n = len(p) n = len(p)
} }
return return
} }
func (c *verifiedConn) WriteBuffer(buffer *buf.Buffer) error {
c.hmacAdd.Write(buffer.Bytes())
dateLen := buffer.Len()
header := buffer.ExtendHeader(tlsHmacHeaderSize)
header[0] = applicationData
header[1] = 3
header[2] = 3
binary.BigEndian.PutUint16(header[3:tlsHeaderSize], hmacSize+uint16(dateLen))
hmacHash := c.hmacAdd.Sum(nil)[:hmacSize]
c.hmacAdd.Write(hmacHash)
copy(header[tlsHeaderSize:], hmacHash)
return c.writer.WriteBuffer(buffer)
}
func (c *verifiedConn) WriteVectorised(buffers []*buf.Buffer) error { func (c *verifiedConn) WriteVectorised(buffers []*buf.Buffer) error {
var header [tlsHmacHeaderSize]byte var header [tlsHmacHeaderSize]byte
header[0] = applicationData header[0] = applicationData
@ -139,7 +155,11 @@ func (c *verifiedConn) WriteVectorised(buffers []*buf.Buffer) error {
} }
c.hmacAdd.Write(c.hmacAdd.Sum(nil)[:hmacSize]) c.hmacAdd.Write(c.hmacAdd.Sum(nil)[:hmacSize])
copy(header[tlsHeaderSize:], c.hmacAdd.Sum(nil)[:hmacSize]) copy(header[tlsHeaderSize:], c.hmacAdd.Sum(nil)[:hmacSize])
return c.writer.WriteVectorised(append([]*buf.Buffer{buf.As(header[:])}, buffers...)) return c.vectorisedWriter.WriteVectorised(append([]*buf.Buffer{buf.As(header[:])}, buffers...))
}
func (c *verifiedConn) FrontHeadroom() int {
return tlsHmacHeaderSize
} }
func (c *verifiedConn) NeedAdditionalReadDeadline() bool { func (c *verifiedConn) NeedAdditionalReadDeadline() bool {