kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use a buffer pool to reduce contentions in tls wrapper

This commit is contained in:
Andy Wang 2020-12-23 00:02:25 +00:00
parent 35f41424c9
commit 4d1612774f
No known key found for this signature in database
GPG Key ID: 181B49F9F38F3374
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
internal/common/tls.go
View File

@ -1,6 +1,7 @@
package common package common
import ( import (
"bytes"
"encoding/binary" "encoding/binary"
"io" "io"
"net" "net"
@ -36,18 +37,15 @@ func AddRecordLayer(input []byte, typ byte, ver uint16) []byte {
type TLSConn struct { type TLSConn struct {
net.Conn net.Conn
writeM sync.Mutex writeBufPool sync.Pool
writeBuf []byte
} }
func NewTLSConn(conn net.Conn) *TLSConn { func NewTLSConn(conn net.Conn) *TLSConn {
writeBuf := make([]byte, initialWriteBufSize)
writeBuf[0] = ApplicationData
writeBuf[1] = byte(VersionTLS13 >> 8)
writeBuf[2] = byte(VersionTLS13 & 0xFF)
return &TLSConn{ return &TLSConn{
Conn: conn, Conn: conn,
writeBuf: writeBuf, writeBufPool: sync.Pool{New: func() interface{} {
return new(bytes.Buffer)
}},
} }
} }
@ -95,13 +93,16 @@ func (tls *TLSConn) Read(buffer []byte) (n int, err error) {
func (tls *TLSConn) Write(in []byte) (n int, err error) { func (tls *TLSConn) Write(in []byte) (n int, err error) {
msgLen := len(in) msgLen := len(in)
tls.writeM.Lock() writeBuf := tls.writeBufPool.Get().(*bytes.Buffer)
tls.writeBuf = append(tls.writeBuf[:5], in...) writeBuf.WriteByte(ApplicationData)
tls.writeBuf[3] = byte(msgLen >> 8) writeBuf.WriteByte(byte(VersionTLS13 >> 8))
tls.writeBuf[4] = byte(msgLen & 0xFF) writeBuf.WriteByte(byte(VersionTLS13 & 0xFF))
n, err = tls.Conn.Write(tls.writeBuf[:recordLayerLength+msgLen]) writeBuf.WriteByte(byte(msgLen >> 8))
tls.writeM.Unlock() writeBuf.WriteByte(byte(msgLen & 0xFF))
return n - recordLayerLength, err writeBuf.Write(in)
i, err := writeBuf.WriteTo(tls.Conn)
tls.writeBufPool.Put(writeBuf)
return int(i - recordLayerLength), err
} }
func (tls *TLSConn) Close() error { func (tls *TLSConn) Close() error {