Archives/Cloak
2
0
Fork 0
mirror of https://github.com/cbeuw/Cloak.git synced 2024-11-03 23:15:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimise AddRecordLayer

Browse Source
This commit is contained in:
Andy Wang 2020-04-10 19:09:17 +01:00
parent e9243a2e9f
commit e4b45a55e3
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
internal/common/tls.go
View File

@ -18,13 +18,19 @@ const (
)
func AddRecordLayer(input []byte, typ byte, ver uint16) []byte {
length := make([]byte, 2)
binary.BigEndian.PutUint16(length, uint16(len(input)))
ret := make([]byte, 5+len(input))
msgLen := len(input)
var ret []byte
if cap(input) > msgLen+recordLayerLength {
ret = input[:msgLen+recordLayerLength]
} else {
ret = make([]byte, msgLen+recordLayerLength)
}
copy(ret[recordLayerLength:], input)
ret[0] = typ
binary.BigEndian.PutUint16(ret[1:3], ver)
copy(ret[3:5], length)
copy(ret[5:], input)
ret[1] = byte(ver >> 8)
ret[2] = byte(ver)
ret[3] = byte(msgLen >> 8)
ret[4] = byte(msgLen)
return ret
}