4d2d91972b
- Add GroupModule to wrap interception, dnsmonitor, and compat modules - Simplify pause/resume operations by grouping related modules - Update worker info collection to handle nested module groups - Remove deprecated flags and improve module lifecycle management - Add proper atomic state tracking for nfqueue interception https://github.com/safing/portmaster/issues/2050
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"runtime"
|
|
|
|
"github.com/maruel/panicparse/v2/stack"
|
|
|
|
"github.com/safing/portmaster/base/utils/debug"
|
|
"github.com/safing/portmaster/service/mgr"
|
|
)
|
|
|
|
// GetWorkerInfo returns the worker info of all running workers.
|
|
func (i *Instance) GetWorkerInfo() (*mgr.WorkerInfo, error) {
|
|
snapshot, _, err := stack.ScanSnapshot(bytes.NewReader(fullStack()), io.Discard, stack.DefaultOpts())
|
|
if err != nil && !errors.Is(err, io.EOF) {
|
|
return nil, fmt.Errorf("get stack: %w", err)
|
|
}
|
|
|
|
infos := make([]*mgr.WorkerInfo, 0, 32)
|
|
for _, m := range i.serviceGroup.Modules() {
|
|
wi, _ := m.Manager().WorkerInfo(snapshot) // Does not fail when we provide a snapshot.
|
|
infos = append(infos, wi)
|
|
|
|
// Check if module is a nested modules group
|
|
if gm, ok := m.(*mgr.GroupModule); ok {
|
|
for _, sm := range gm.Modules() {
|
|
wi, _ := sm.Manager().WorkerInfo(snapshot) // Does not fail when we provide a snapshot.
|
|
infos = append(infos, wi)
|
|
}
|
|
}
|
|
}
|
|
for _, m := range i.SpnGroup.Modules() {
|
|
wi, _ := m.Manager().WorkerInfo(snapshot) // Does not fail when we provide a snapshot.
|
|
infos = append(infos, wi)
|
|
|
|
// Check if module is a nested modules group
|
|
if gm, ok := m.(*mgr.GroupModule); ok {
|
|
for _, sm := range gm.Modules() {
|
|
wi, _ := sm.Manager().WorkerInfo(snapshot) // Does not fail when we provide a snapshot.
|
|
infos = append(infos, wi)
|
|
}
|
|
}
|
|
}
|
|
|
|
return mgr.MergeWorkerInfo(infos...), nil
|
|
}
|
|
|
|
// AddWorkerInfoToDebugInfo adds the worker info of all running workers to the debug info.
|
|
func (i *Instance) AddWorkerInfoToDebugInfo(di *debug.Info) {
|
|
info, err := i.GetWorkerInfo()
|
|
if err != nil {
|
|
di.AddSection(
|
|
"Worker Status Failed",
|
|
debug.UseCodeSection,
|
|
err.Error(),
|
|
)
|
|
return
|
|
}
|
|
|
|
di.AddSection(
|
|
fmt.Sprintf("Worker Status: %d/%d (%d?)", info.Running, len(info.Workers), info.Missing+info.Other),
|
|
debug.UseCodeSection,
|
|
info.Format(),
|
|
)
|
|
}
|
|
|
|
func fullStack() []byte {
|
|
buf := make([]byte, 8096)
|
|
for {
|
|
n := runtime.Stack(buf, true)
|
|
if n < len(buf) {
|
|
return buf[:n]
|
|
}
|
|
buf = make([]byte, 2*len(buf))
|
|
}
|
|
}
|