92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
/* SPDX-License-Identifier: MIT
|
|
*
|
|
* Copyright (C) 2017-2021 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
package device
|
|
|
|
import "sync"
|
|
|
|
func (device *Device) PopulatePools() {
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
device.pool.messageBufferPool = &sync.Pool{
|
|
New: func() interface{} {
|
|
return new([MaxMessageSize]byte)
|
|
},
|
|
}
|
|
device.pool.inboundElementPool = &sync.Pool{
|
|
New: func() interface{} {
|
|
return new(QueueInboundElement)
|
|
},
|
|
}
|
|
device.pool.outboundElementPool = &sync.Pool{
|
|
New: func() interface{} {
|
|
return new(QueueOutboundElement)
|
|
},
|
|
}
|
|
} else {
|
|
device.pool.messageBufferReuseChan = make(chan *[MaxMessageSize]byte, PreallocatedBuffersPerPool)
|
|
for i := 0; i < PreallocatedBuffersPerPool; i++ {
|
|
device.pool.messageBufferReuseChan <- new([MaxMessageSize]byte)
|
|
}
|
|
device.pool.inboundElementReuseChan = make(chan *QueueInboundElement, PreallocatedBuffersPerPool)
|
|
for i := 0; i < PreallocatedBuffersPerPool; i++ {
|
|
device.pool.inboundElementReuseChan <- new(QueueInboundElement)
|
|
}
|
|
device.pool.outboundElementReuseChan = make(chan *QueueOutboundElement, PreallocatedBuffersPerPool)
|
|
for i := 0; i < PreallocatedBuffersPerPool; i++ {
|
|
device.pool.outboundElementReuseChan <- new(QueueOutboundElement)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (device *Device) GetMessageBuffer() *[MaxMessageSize]byte {
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
return device.pool.messageBufferPool.Get().(*[MaxMessageSize]byte)
|
|
} else {
|
|
return <-device.pool.messageBufferReuseChan
|
|
}
|
|
}
|
|
|
|
func (device *Device) PutMessageBuffer(msg *[MaxMessageSize]byte) {
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
device.pool.messageBufferPool.Put(msg)
|
|
} else {
|
|
device.pool.messageBufferReuseChan <- msg
|
|
}
|
|
}
|
|
|
|
func (device *Device) GetInboundElement() *QueueInboundElement {
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
return device.pool.inboundElementPool.Get().(*QueueInboundElement)
|
|
} else {
|
|
return <-device.pool.inboundElementReuseChan
|
|
}
|
|
}
|
|
|
|
func (device *Device) PutInboundElement(elem *QueueInboundElement) {
|
|
elem.clearPointers()
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
device.pool.inboundElementPool.Put(elem)
|
|
} else {
|
|
device.pool.inboundElementReuseChan <- elem
|
|
}
|
|
}
|
|
|
|
func (device *Device) GetOutboundElement() *QueueOutboundElement {
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
return device.pool.outboundElementPool.Get().(*QueueOutboundElement)
|
|
} else {
|
|
return <-device.pool.outboundElementReuseChan
|
|
}
|
|
}
|
|
|
|
func (device *Device) PutOutboundElement(elem *QueueOutboundElement) {
|
|
elem.clearPointers()
|
|
if PreallocatedBuffersPerPool == 0 {
|
|
device.pool.outboundElementPool.Put(elem)
|
|
} else {
|
|
device.pool.outboundElementReuseChan <- elem
|
|
}
|
|
}
|