mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
39 lines
1 KiB
Go
39 lines
1 KiB
Go
package state
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/safing/portmaster/network/proc"
|
|
"github.com/safing/portmaster/network/socket"
|
|
)
|
|
|
|
var (
|
|
getTCP4Table = proc.GetTCP4Table
|
|
getTCP6Table = proc.GetTCP6Table
|
|
getUDP4Table = proc.GetUDP4Table
|
|
getUDP6Table = proc.GetUDP6Table
|
|
)
|
|
|
|
var baseWaitTime = 3 * 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++ {
|
|
// look for PID
|
|
pid = proc.GetPID(socketInfo)
|
|
if pid != socket.UndefinedProcessID {
|
|
// if we found a PID, return
|
|
break
|
|
}
|
|
|
|
// every time, except for the last iteration
|
|
if i < lookupTries {
|
|
// 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)
|
|
}
|
|
}
|
|
|
|
return pid, connInbound, nil
|
|
}
|