Reduce PID finding retries and simply proc interface on linux

This commit is contained in:
Daniel
2023-12-18 17:03:48 +01:00
parent 6822b7a8a8
commit d67b1e8a64
5 changed files with 24 additions and 72 deletions

View File

@@ -28,6 +28,11 @@ var (
ErrPIDNotFound = errors.New("could not find pid for socket inode")
)
const (
lookupTries = 5
fastLookupTries = 2
)
// Lookup looks for the given connection in the system state tables and returns the PID of the associated process and whether the connection is inbound.
func Lookup(pktInfo *packet.Info, fast bool) (pid int, inbound bool, err error) {
// auto-detect version

View File

@@ -10,11 +10,6 @@ import (
"github.com/safing/portmaster/network/socket"
)
var (
lookupTries = 20 // With a max wait of 5ms, this amounts to up to 100ms.
fastLookupTries = 2
)
func init() {
// This increases performance on unsupported system.
// It's not critical at all and does not break anything if it fails.

View File

@@ -13,16 +13,14 @@ var (
getUDP4Table = proc.GetUDP4Table
getUDP6Table = proc.GetUDP6Table
lookupTries = 20 // With a max wait of 5ms, this amounts to up to 100ms.
fastLookupTries = 2
baseWaitTime = 3 * time.Millisecond
checkPIDTries = 5
checkPIDBaseWaitTime = 5 * time.Millisecond
)
// CheckPID checks the if socket info already has a PID and if not, tries to find it.
// Depending on the OS, this might be a no-op.
func CheckPID(socketInfo socket.Info, connInbound bool) (pid int, inbound bool, err error) {
for i := 1; i <= lookupTries; i++ {
for i := 1; i <= checkPIDTries; i++ {
// look for PID
pid = proc.GetPID(socketInfo)
if pid != socket.UndefinedProcessID {
@@ -31,10 +29,10 @@ func CheckPID(socketInfo socket.Info, connInbound bool) (pid int, inbound bool,
}
// every time, except for the last iteration
if i < lookupTries {
if i < checkPIDTries {
// we found no PID, we could have been too fast, give the kernel some time to think
// back off timer: with 3ms baseWaitTime: 3, 6, 9, 12, 15, 18, 21ms - 84ms in total
time.Sleep(time.Duration(i+1) * baseWaitTime)
// back off timer: with 5ms baseWaitTime: 5, 10, 15, 20, 25 - 75ms in total
time.Sleep(time.Duration(i) * checkPIDBaseWaitTime)
}
}

View File

@@ -10,13 +10,6 @@ var (
getTCP6Table = iphelper.GetTCP6Table
getUDP4Table = iphelper.GetUDP4Table
getUDP6Table = iphelper.GetUDP6Table
// With a max wait of 5ms, this amounts to up to 25ms,
// excluding potential data fetching time.
// Measured on Windows: ~150ms
lookupTries = 5
fastLookupTries = 2
)
// CheckPID checks the if socket info already has a PID and if not, tries to find it.