mirror of
https://github.com/safing/portmaster
synced 2025-09-01 01:59:11 +00:00
46 lines
878 B
Go
46 lines
878 B
Go
// Copyright Safing ICS Technologies GmbH. Use of this source code is governed by the AGPL license that can be found in the LICENSE file.
|
|
|
|
package nfqueue
|
|
|
|
import (
|
|
"github.com/Safing/safing-core/network/packet"
|
|
"sync"
|
|
)
|
|
|
|
type multiQueue struct {
|
|
qs []*nfQueue
|
|
}
|
|
|
|
func NewMultiQueue(min, max uint16) (mq *multiQueue) {
|
|
mq = &multiQueue{make([]*nfQueue, 0, max-min)}
|
|
for i := min; i < max; i++ {
|
|
mq.qs = append(mq.qs, NewNFQueue(i))
|
|
}
|
|
return mq
|
|
}
|
|
|
|
func (mq *multiQueue) Process() <-chan packet.Packet {
|
|
var (
|
|
wg sync.WaitGroup
|
|
out = make(chan packet.Packet, len(mq.qs))
|
|
)
|
|
for _, q := range mq.qs {
|
|
wg.Add(1)
|
|
go func(ch <-chan packet.Packet) {
|
|
for pkt := range ch {
|
|
out <- pkt
|
|
}
|
|
wg.Done()
|
|
}(q.Process())
|
|
}
|
|
go func() {
|
|
wg.Wait()
|
|
close(out)
|
|
}()
|
|
return out
|
|
}
|
|
func (mq *multiQueue) Destroy() {
|
|
for _, q := range mq.qs {
|
|
q.Destroy()
|
|
}
|
|
}
|