mirror of
https://github.com/ScuroNeko/mtg.git
synced 2026-08-31 21:34:02 +03:00
Cleanup pools from wrappes
This commit is contained in:
@@ -39,13 +39,11 @@ func (w *wrapperFakeTLS) WriteTimeout(p []byte, timeout time.Duration) (int, err
|
||||
|
||||
func (w *wrapperFakeTLS) write(p []byte, writeFunc func([]byte) (int, error)) (int, error) {
|
||||
sum := 0
|
||||
|
||||
buf := acquireBytesBuffer()
|
||||
defer releaseBytesBuffer(buf)
|
||||
buf := bytes.Buffer{}
|
||||
|
||||
for _, v := range tlstypes.MakeRecords(p) {
|
||||
buf.Reset()
|
||||
v.WriteBytes(buf)
|
||||
v.WriteBytes(&buf)
|
||||
|
||||
_, err := writeFunc(buf.Bytes())
|
||||
if err != nil {
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package stream
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/aes"
|
||||
"crypto/cipher"
|
||||
"crypto/md5" // nolint: gosec
|
||||
@@ -53,8 +54,8 @@ func mtprotoDeriveKeys(purpose mtprotoCipherPurpose,
|
||||
resp *rpc.NonceResponse,
|
||||
client, remote *net.TCPAddr,
|
||||
secret []byte) ([]byte, []byte) {
|
||||
message := acquireBytesBuffer()
|
||||
defer releaseBytesBuffer(message)
|
||||
|
||||
message := bytes.Buffer{}
|
||||
|
||||
message.Write(resp.Nonce) // nolint: gosec
|
||||
message.Write(req.Nonce) // nolint: gosec
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package stream
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/cipher"
|
||||
"fmt"
|
||||
"net"
|
||||
@@ -40,8 +41,7 @@ func (w *wrapperObfuscated2) Read(p []byte) (int, error) {
|
||||
}
|
||||
|
||||
func (w *wrapperObfuscated2) WriteTimeout(p []byte, timeout time.Duration) (int, error) {
|
||||
buffer := acquireBytesBuffer()
|
||||
defer releaseBytesBuffer(buffer)
|
||||
buffer := bytes.Buffer{}
|
||||
|
||||
buffer.Write(p)
|
||||
|
||||
@@ -53,8 +53,7 @@ func (w *wrapperObfuscated2) WriteTimeout(p []byte, timeout time.Duration) (int,
|
||||
}
|
||||
|
||||
func (w *wrapperObfuscated2) Write(p []byte) (int, error) {
|
||||
buffer := acquireBytesBuffer()
|
||||
defer releaseBytesBuffer(buffer)
|
||||
buffer := bytes.Buffer{}
|
||||
|
||||
buffer.Write(p)
|
||||
|
||||
|
||||
@@ -1,23 +0,0 @@
|
||||
package stream
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"sync"
|
||||
)
|
||||
|
||||
var (
|
||||
poolBytesBuffer = sync.Pool{
|
||||
New: func() interface{} {
|
||||
return &bytes.Buffer{}
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
func acquireBytesBuffer() *bytes.Buffer {
|
||||
return poolBytesBuffer.Get().(*bytes.Buffer)
|
||||
}
|
||||
|
||||
func releaseBytesBuffer(buf *bytes.Buffer) {
|
||||
buf.Reset()
|
||||
poolBytesBuffer.Put(buf)
|
||||
}
|
||||
Reference in New Issue
Block a user