Move interception module and better integrate workers

This commit is contained in:
Daniel
2023-07-20 13:43:55 +02:00
parent 41c5266315
commit ec85816577
7 changed files with 210 additions and 122 deletions

View File

@@ -1,7 +1,9 @@
package interception
import (
"context"
"fmt"
"time"
"github.com/safing/portmaster/firewall/interception/windowskext"
"github.com/safing/portmaster/network"
@@ -10,7 +12,7 @@ import (
)
// start starts the interception.
func start(ch chan packet.Packet) error {
func startInterception(packets chan packet.Packet) error {
kextFile, err := updates.GetPlatformFile("kext/portmaster-kext.sys")
if err != nil {
return fmt.Errorf("interception: could not get kext sys: %s", err)
@@ -26,16 +28,22 @@ func start(ch chan packet.Packet) error {
return fmt.Errorf("interception: could not start windows kext: %s", err)
}
go windowskext.Handler(ch)
// Start packet handler.
module.StartServiceWorker("kext packet handler", 0, func(ctx context.Context) error {
windowskext.Handler(ctx, packets)
return nil
})
// Example worker for the bandwidth data stats. Not ment for production.
// windowskext.StartBandwidthWorker()
// Start bandwidth stats monitor.
module.StartServiceWorker("kext bandwidth stats monitor", 0, func(ctx context.Context) error {
return windowskext.BandwidthStatsWorker(ctx, 1*time.Second, BandwidthUpdates)
})
return nil
}
// stop starts the interception.
func stop() error {
func stopInterception() error {
return windowskext.Stop()
}