mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
Improve support for disabled IPv6 stack
This commit is contained in:
parent
fa98f1ea6d
commit
4b29eed2cf
4 changed files with 40 additions and 15 deletions
|
@ -2,6 +2,8 @@ package network
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/safing/portbase/modules"
|
"github.com/safing/portbase/modules"
|
||||||
|
"github.com/safing/portmaster/netenv"
|
||||||
|
"github.com/safing/portmaster/network/state"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -11,7 +13,7 @@ var (
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
module = modules.Register("network", prep, start, nil, "base", "processes")
|
module = modules.Register("network", prep, start, nil, "base", "netenv", "processes")
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetDefaultFirewallHandler sets the default firewall handler.
|
// SetDefaultFirewallHandler sets the default firewall handler.
|
||||||
|
@ -22,6 +24,11 @@ func SetDefaultFirewallHandler(handler FirewallHandler) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func prep() error {
|
func prep() error {
|
||||||
|
if netenv.IPv6Enabled() {
|
||||||
|
state.EnableTCPDualStack()
|
||||||
|
state.EnableUDPDualStack()
|
||||||
|
}
|
||||||
|
|
||||||
return registerAPIEndpoints()
|
return registerAPIEndpoints()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -3,6 +3,8 @@ package state
|
||||||
import (
|
import (
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
|
"github.com/safing/portmaster/netenv"
|
||||||
|
|
||||||
"github.com/safing/portbase/database/record"
|
"github.com/safing/portbase/database/record"
|
||||||
"github.com/safing/portmaster/network/socket"
|
"github.com/safing/portmaster/network/socket"
|
||||||
)
|
)
|
||||||
|
@ -30,21 +32,23 @@ func GetInfo() *Info {
|
||||||
info.TCP4Listeners = tcp4Table.listeners
|
info.TCP4Listeners = tcp4Table.listeners
|
||||||
tcp4Table.lock.RUnlock()
|
tcp4Table.lock.RUnlock()
|
||||||
|
|
||||||
|
udp4Table.updateTable()
|
||||||
|
udp4Table.lock.RLock()
|
||||||
|
info.UDP4Binds = udp4Table.binds
|
||||||
|
udp4Table.lock.RUnlock()
|
||||||
|
|
||||||
|
if netenv.IPv6Enabled() {
|
||||||
tcp6Table.updateTables()
|
tcp6Table.updateTables()
|
||||||
tcp6Table.lock.RLock()
|
tcp6Table.lock.RLock()
|
||||||
info.TCP6Connections = tcp6Table.connections
|
info.TCP6Connections = tcp6Table.connections
|
||||||
info.TCP6Listeners = tcp6Table.listeners
|
info.TCP6Listeners = tcp6Table.listeners
|
||||||
tcp6Table.lock.RUnlock()
|
tcp6Table.lock.RUnlock()
|
||||||
|
|
||||||
udp4Table.updateTable()
|
|
||||||
udp4Table.lock.RLock()
|
|
||||||
info.UDP4Binds = udp4Table.binds
|
|
||||||
udp4Table.lock.RUnlock()
|
|
||||||
|
|
||||||
udp6Table.updateTable()
|
udp6Table.updateTable()
|
||||||
udp6Table.lock.RLock()
|
udp6Table.lock.RLock()
|
||||||
info.UDP6Binds = udp6Table.binds
|
info.UDP6Binds = udp6Table.binds
|
||||||
udp6Table.lock.RUnlock()
|
udp6Table.lock.RUnlock()
|
||||||
|
}
|
||||||
|
|
||||||
info.UpdateMeta()
|
info.UpdateMeta()
|
||||||
return info
|
return info
|
||||||
|
|
|
@ -29,6 +29,11 @@ var (
|
||||||
tcp4Table = &tcpTable{
|
tcp4Table = &tcpTable{
|
||||||
version: 4,
|
version: 4,
|
||||||
fetchTable: getTCP4Table,
|
fetchTable: getTCP4Table,
|
||||||
dualStack: tcp6Table,
|
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// EnableTCPDualStack adds the TCP6 table to the TCP4 table as a dual-stack.
|
||||||
|
// Must be called before any lookup operation.
|
||||||
|
func EnableTCPDualStack() {
|
||||||
|
tcp4Table.dualStack = tcp6Table
|
||||||
|
}
|
||||||
|
|
|
@ -6,6 +6,8 @@ import (
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/safing/portmaster/netenv"
|
||||||
|
|
||||||
"github.com/safing/portbase/utils"
|
"github.com/safing/portbase/utils"
|
||||||
"github.com/safing/portmaster/network/packet"
|
"github.com/safing/portmaster/network/packet"
|
||||||
"github.com/safing/portmaster/network/socket"
|
"github.com/safing/portmaster/network/socket"
|
||||||
|
@ -53,10 +55,15 @@ var (
|
||||||
version: 4,
|
version: 4,
|
||||||
fetchTable: getUDP4Table,
|
fetchTable: getUDP4Table,
|
||||||
states: make(map[string]map[string]*udpState),
|
states: make(map[string]map[string]*udpState),
|
||||||
dualStack: udp6Table,
|
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// EnableUDPDualStack adds the UDP6 table to the UDP4 table as a dual-stack.
|
||||||
|
// Must be called before any lookup operation.
|
||||||
|
func EnableUDPDualStack() {
|
||||||
|
udp4Table.dualStack = udp6Table
|
||||||
|
}
|
||||||
|
|
||||||
// CleanUDPStates cleans the udp connection states which save connection directions.
|
// CleanUDPStates cleans the udp connection states which save connection directions.
|
||||||
func CleanUDPStates(_ context.Context) {
|
func CleanUDPStates(_ context.Context) {
|
||||||
now := time.Now().UTC()
|
now := time.Now().UTC()
|
||||||
|
@ -64,8 +71,10 @@ func CleanUDPStates(_ context.Context) {
|
||||||
udp4Table.updateTable()
|
udp4Table.updateTable()
|
||||||
udp4Table.cleanStates(now)
|
udp4Table.cleanStates(now)
|
||||||
|
|
||||||
|
if netenv.IPv6Enabled() {
|
||||||
udp6Table.updateTable()
|
udp6Table.updateTable()
|
||||||
udp6Table.cleanStates(now)
|
udp6Table.cleanStates(now)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (table *udpTable) getConnState(
|
func (table *udpTable) getConnState(
|
||||||
|
|
Loading…
Add table
Reference in a new issue