diff --git a/faketls/client_protocol.go b/faketls/client_protocol.go index 135c422..08e896d 100644 --- a/faketls/client_protocol.go +++ b/faketls/client_protocol.go @@ -2,6 +2,7 @@ package faketls import ( "bufio" + "bytes" "encoding/binary" "errors" "fmt" @@ -63,9 +64,7 @@ func (c *ClientProtocol) tlsHandshake(conn io.ReadWriter) error { return fmt.Errorf("cannot read initial record: %w", err) } - buf := acquireBytesBuffer() - defer releaseBytesBuffer(buf) - + buf := &bytes.Buffer{} helloRecord.Data.WriteBytes(buf) clientHello, err := tlstypes.ParseClientHello(buf.Bytes()) diff --git a/faketls/cloak.go b/faketls/cloak.go index d9a2cc2..1e3a5b0 100644 --- a/faketls/cloak.go +++ b/faketls/cloak.go @@ -67,8 +67,5 @@ func cloak(one, another io.ReadWriteCloser) { func cloakPipe(one io.Writer, another io.Reader, wg *sync.WaitGroup) { defer wg.Done() - buf := acquireCloakBuffer() - defer releaseCloakBuffer(buf) - - io.CopyBuffer(one, another, *buf) // nolint: errcheck + io.Copy(one, another) // nolint: errcheck } diff --git a/faketls/pools.go b/faketls/pools.go deleted file mode 100644 index ca7b6c4..0000000 --- a/faketls/pools.go +++ /dev/null @@ -1,39 +0,0 @@ -package faketls - -import ( - "bytes" - "sync" -) - -const cloakBufferSize = 1024 - -var ( - poolBytesBuffer = sync.Pool{ - New: func() interface{} { - return &bytes.Buffer{} - }, - } - poolCloakBuffer = sync.Pool{ - New: func() interface{} { - rv := make([]byte, cloakBufferSize) - return &rv - }, - } -) - -func acquireBytesBuffer() *bytes.Buffer { - return poolBytesBuffer.Get().(*bytes.Buffer) -} - -func acquireCloakBuffer() *[]byte { - return poolCloakBuffer.Get().(*[]byte) -} - -func releaseBytesBuffer(buf *bytes.Buffer) { - buf.Reset() - poolBytesBuffer.Put(buf) -} - -func releaseCloakBuffer(buf *[]byte) { - poolCloakBuffer.Put(buf) -}