make putting back packet buffers a member function of packetBuffer

This commit is contained in:
Marten Seemann
2018-12-30 15:28:52 +07:00
parent d3ea7c0c4c
commit b32fb438af
5 changed files with 35 additions and 32 deletions

View File

@@ -479,7 +479,7 @@ func (s *session) handlePacketImpl(p *receivedPacket) bool /* was the packet suc
defer func() {
// Put back the packet buffer if the packet wasn't queued for later decryption.
if !wasQueued {
putPacketBuffer(p.buffer)
p.buffer.Release()
}
}()
@@ -962,7 +962,7 @@ func (s *session) sendPacket() (bool, error) {
}
func (s *session) sendPackedPacket(packet *packedPacket) error {
defer putPacketBuffer(packet.buffer)
defer packet.buffer.Release()
s.logPacket(packet)
return s.conn.Write(packet.raw)
}