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

Use pooled buffer for session closing frame

This commit is contained in:
Andy Wang 2020-12-24 13:42:22 +00:00
parent 881f6e6f9d
commit 4f34e69006
No known key found for this signature in database
GPG Key ID: 181B49F9F38F3374
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
internal/multiplex/session.go
View File

@ -319,10 +319,10 @@ func (sesh *Session) Close() error {
}
// we send a notice frame telling remote to close the session
padBuf := make([]byte, 256+frameHeaderLength+sesh.Obfuscator.maxOverhead)
common.CryptoRandRead(padBuf[:1])
padLen := int(padBuf[0]) + 1
payload := padBuf[frameHeaderLength : padLen+frameHeaderLength]
buf := sesh.streamObfsBufPool.Get().(*[]byte)
common.CryptoRandRead((*buf)[:1])
padLen := int((*buf)[0]) + 1
payload := (*buf)[frameHeaderLength : padLen+frameHeaderLength]
common.CryptoRandRead(payload)
f := &Frame{
@ -331,11 +331,11 @@ func (sesh *Session) Close() error {
Closing: closingSession,
Payload: payload,
}
i, err := sesh.Obfs(f, padBuf, frameHeaderLength)
i, err := sesh.Obfs(f, *buf, frameHeaderLength)
if err != nil {
return err
}
_, err = sesh.sb.send(padBuf[:i], new(uint32))
_, err = sesh.sb.send((*buf)[:i], new(uint32))
if err != nil {
return err
}