mirror of
https://github.com/safing/portmaster
synced 2025-09-02 10:39:22 +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>
81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
//go:build linux
|
|
|
|
package proc
|
|
|
|
import (
|
|
"errors"
|
|
"io/fs"
|
|
"os"
|
|
"strconv"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/safing/portmaster/base/log"
|
|
"github.com/safing/portmaster/base/utils"
|
|
)
|
|
|
|
var (
|
|
// pidsByUserLock is also used for locking the socketInfo.PID on all socket.*Info structs.
|
|
pidsByUser = make(map[int][]int)
|
|
pidsByUserLock sync.RWMutex
|
|
fetchPidsByUser = utils.NewCallLimiter(10 * time.Millisecond)
|
|
)
|
|
|
|
// getPidsByUser returns the cached PIDs for the given UID.
|
|
func getPidsByUser(uid int) (pids []int, ok bool) {
|
|
pidsByUserLock.RLock()
|
|
defer pidsByUserLock.RUnlock()
|
|
|
|
pids, ok = pidsByUser[uid]
|
|
return
|
|
}
|
|
|
|
// updatePids fetches and creates a new pidsByUser map using a call limiter.
|
|
func updatePids() {
|
|
fetchPidsByUser.Do(func() {
|
|
newPidsByUser := make(map[int][]int)
|
|
pidCnt := 0
|
|
|
|
entries := readDirNames("/proc")
|
|
if len(entries) == 0 {
|
|
log.Warning("proc: found no PIDs in /proc")
|
|
return
|
|
}
|
|
|
|
entryLoop:
|
|
for _, entry := range entries {
|
|
pid, err := strconv.ParseInt(entry, 10, 32)
|
|
if err != nil {
|
|
continue entryLoop
|
|
}
|
|
|
|
statData, err := os.Stat("/proc/" + strconv.FormatInt(pid, 10))
|
|
if err != nil {
|
|
if !errors.Is(err, fs.ErrNotExist) {
|
|
log.Warningf("proc: could not stat /proc/%d: %s", pid, err)
|
|
}
|
|
continue entryLoop
|
|
}
|
|
sys, ok := statData.Sys().(*syscall.Stat_t)
|
|
if !ok {
|
|
log.Warningf("proc: unable to parse /proc/%d: wrong type", pid)
|
|
continue entryLoop
|
|
}
|
|
|
|
pids, ok := newPidsByUser[int(sys.Uid)]
|
|
if ok {
|
|
newPidsByUser[int(sys.Uid)] = append(pids, int(pid))
|
|
} else {
|
|
newPidsByUser[int(sys.Uid)] = []int{int(pid)}
|
|
}
|
|
pidCnt++
|
|
}
|
|
|
|
// log.Tracef("proc: updated PID table with %d entries", pidCnt)
|
|
|
|
pidsByUserLock.Lock()
|
|
defer pidsByUserLock.Unlock()
|
|
pidsByUser = newPidsByUser
|
|
})
|
|
}
|