Fix a bug with crashing desktop telegram

This commit is contained in:
9seconds
2020-03-24 16:36:03 +03:00
parent 391d901402
commit bc72717582
2 changed files with 2 additions and 17 deletions
-14
View File
@@ -11,27 +11,13 @@ var (
return &bytes.Buffer{}
},
}
poolProxyBytesBuffer = sync.Pool{
New: func() interface{} {
return &bytes.Buffer{}
},
}
)
func acquireClientBytesBuffer() *bytes.Buffer {
return poolClientBytesBuffer.Get().(*bytes.Buffer)
}
func acquireProxyBytesBuffer() *bytes.Buffer {
return poolProxyBytesBuffer.Get().(*bytes.Buffer)
}
func releaseClientBytesBuffer(buf *bytes.Buffer) {
buf.Reset()
poolClientBytesBuffer.Put(buf)
}
func releaseProxyBytesBuffer(buf *bytes.Buffer) {
buf.Reset()
poolProxyBytesBuffer.Put(buf)
}
+2 -3
View File
@@ -22,10 +22,9 @@ type wrapperProxy struct {
}
func (w *wrapperProxy) Write(packet conntypes.Packet, acks *conntypes.ConnectionAcks) error {
buf := acquireProxyBytesBuffer()
defer releaseProxyBytesBuffer(buf)
buf := bytes.Buffer{}
flags := w.flags
if acks.Quick {
flags |= rpc.ProxyRequestFlagsQuickAck
}