ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-09-01 15:44:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixes

Browse Source
This commit is contained in:
9seconds
2020-03-23 10:53:58 +03:00
parent d72ed03a43
commit 6a9069ca72
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wrappers/stream/obfuscated2.go
+10 -9
View File
@@ -21,11 +21,8 @@ var (
} }
) )
func poolWrapperObfuscated2WritePoolAcquire(size int) *bytes.Buffer { func poolWrapperObfuscated2WritePoolAcquire() *bytes.Buffer {
buf := poolWrapperObfuscated2WritePool.Get().(*bytes.Buffer) return poolWrapperObfuscated2WritePool.Get().(*bytes.Buffer)
buf.Grow(size)
return buf
} }
func poolWrapperObfuscated2WritePoolRelease(buf *bytes.Buffer) { func poolWrapperObfuscated2WritePoolRelease(buf *bytes.Buffer) {
@@ -62,22 +59,26 @@ func (w *wrapperObfuscated2) Read(p []byte) (int, error) {
} }
func (w *wrapperObfuscated2) WriteTimeout(p []byte, timeout time.Duration) (int, error) { func (w *wrapperObfuscated2) WriteTimeout(p []byte, timeout time.Duration) (int, error) {
buffer := poolWrapperObfuscated2WritePoolAcquire(len(p)) buffer := poolWrapperObfuscated2WritePoolAcquire()
defer poolWrapperObfuscated2WritePoolRelease(buffer) defer poolWrapperObfuscated2WritePoolRelease(buffer)
buffer.Write(p)
buf := buffer.Bytes() buf := buffer.Bytes()
copy(buf, p)
w.encryptor.XORKeyStream(buf, buf) w.encryptor.XORKeyStream(buf, buf)
return w.parent.WriteTimeout(buf, timeout) return w.parent.WriteTimeout(buf, timeout)
} }
func (w *wrapperObfuscated2) Write(p []byte) (int, error) { func (w *wrapperObfuscated2) Write(p []byte) (int, error) {
buffer := poolWrapperObfuscated2WritePoolAcquire(len(p)) buffer := poolWrapperObfuscated2WritePoolAcquire()
defer poolWrapperObfuscated2WritePoolRelease(buffer) defer poolWrapperObfuscated2WritePoolRelease(buffer)
buffer.Write(p)
buf := buffer.Bytes() buf := buffer.Bytes()
copy(buf, p)
w.encryptor.XORKeyStream(buf, buf) w.encryptor.XORKeyStream(buf, buf)
return w.parent.Write(buf) return w.parent.Write(buf)