mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
80664d1a27
* Move portbase into monorepo * Add new simple module mgr * [WIP] Switch to new simple module mgr * Add StateMgr and more worker variants * [WIP] Switch more modules * [WIP] Switch more modules * [WIP] swtich more modules * [WIP] switch all SPN modules * [WIP] switch all service modules * [WIP] Convert all workers to the new module system * [WIP] add new task system to module manager * [WIP] Add second take for scheduling workers * [WIP] Add FIXME for bugs in new scheduler * [WIP] Add minor improvements to scheduler * [WIP] Add new worker scheduler * [WIP] Fix more bug related to new module system * [WIP] Fix start handing of the new module system * [WIP] Improve startup process * [WIP] Fix minor issues * [WIP] Fix missing subsystem in settings * [WIP] Initialize managers in constructor * [WIP] Move module event initialization to constrictors * [WIP] Fix setting for enabling and disabling the SPN module * [WIP] Move API registeration into module construction * [WIP] Update states mgr for all modules * [WIP] Add CmdLine operation support * Add state helper methods to module group and instance * Add notification and module status handling to status package * Fix starting issues * Remove pilot widget and update security lock to new status data * Remove debug logs * Improve http server shutdown * Add workaround for cleanly shutting down firewall+netquery * Improve logging * Add syncing states with notifications for new module system * Improve starting, stopping, shutdown; resolve FIXMEs/TODOs * [WIP] Fix most unit tests * Review new module system and fix minor issues * Push shutdown and restart events again via API * Set sleep mode via interface * Update example/template module * [WIP] Fix spn/cabin unit test * Remove deprecated UI elements * Make log output more similar for the logging transition phase * Switch spn hub and observer cmds to new module system * Fix log sources * Make worker mgr less error prone * Fix tests and minor issues * Fix observation hub * Improve shutdown and restart handling * Split up big connection.go source file * Move varint and dsd packages to structures repo * Improve expansion test * Fix linter warnings * Fix interception module on windows * Fix linter errors --------- Co-authored-by: Vladimir Stoilov <vladimir@safing.io>
91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
package state
|
|
|
|
import (
|
|
"net"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/safing/portmaster/base/log"
|
|
"github.com/safing/portmaster/base/utils"
|
|
"github.com/safing/portmaster/service/network/socket"
|
|
)
|
|
|
|
const (
|
|
minDurationBetweenTableUpdates = 10 * time.Millisecond
|
|
)
|
|
|
|
type tcpTable struct {
|
|
version int
|
|
|
|
connections []*socket.ConnectionInfo
|
|
listeners []*socket.BindInfo
|
|
lock sync.RWMutex
|
|
|
|
// lastUpdateAt stores the time when the tables where last updated as unix nanoseconds.
|
|
lastUpdateAt atomic.Int64
|
|
|
|
fetchLimiter *utils.CallLimiter
|
|
fetchTable func() (connections []*socket.ConnectionInfo, listeners []*socket.BindInfo, err error)
|
|
|
|
dualStack *tcpTable
|
|
}
|
|
|
|
var (
|
|
tcp6Table = &tcpTable{
|
|
version: 6,
|
|
fetchLimiter: utils.NewCallLimiter(minDurationBetweenTableUpdates),
|
|
fetchTable: getTCP6Table,
|
|
}
|
|
|
|
tcp4Table = &tcpTable{
|
|
version: 4,
|
|
fetchLimiter: utils.NewCallLimiter(minDurationBetweenTableUpdates),
|
|
fetchTable: getTCP4Table,
|
|
}
|
|
)
|
|
|
|
// 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
|
|
}
|
|
|
|
func (table *tcpTable) getCurrentTables() (
|
|
connections []*socket.ConnectionInfo,
|
|
listeners []*socket.BindInfo,
|
|
) {
|
|
table.lock.RLock()
|
|
defer table.lock.RUnlock()
|
|
|
|
return table.connections, table.listeners
|
|
}
|
|
|
|
func (table *tcpTable) updateTables() (
|
|
connections []*socket.ConnectionInfo,
|
|
listeners []*socket.BindInfo,
|
|
) {
|
|
// Fetch tables.
|
|
table.fetchLimiter.Do(func() {
|
|
// Fetch new tables from system.
|
|
connections, listeners, err := table.fetchTable()
|
|
if err != nil {
|
|
log.Warningf("state: failed to get TCP%d socket table: %s", table.version, err)
|
|
return
|
|
}
|
|
|
|
// Pre-check for any listeners.
|
|
for _, bindInfo := range listeners {
|
|
bindInfo.ListensAny = bindInfo.Local.IP.Equal(net.IPv4zero) || bindInfo.Local.IP.Equal(net.IPv6zero)
|
|
}
|
|
|
|
// Apply new tables.
|
|
table.lock.Lock()
|
|
defer table.lock.Unlock()
|
|
table.connections = connections
|
|
table.listeners = listeners
|
|
table.lastUpdateAt.Store(time.Now().UnixNano())
|
|
})
|
|
|
|
return table.getCurrentTables()
|
|
}
|