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>
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package core
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
|
|
"github.com/safing/portmaster/base/database"
|
|
"github.com/safing/portmaster/base/database/record"
|
|
"github.com/safing/portmaster/base/runtime"
|
|
"github.com/safing/portmaster/service/mgr"
|
|
)
|
|
|
|
var modulesIntegrationUpdatePusher func(...record.Record)
|
|
|
|
func initModulesIntegration() (err error) {
|
|
modulesIntegrationUpdatePusher, err = runtime.Register("modules/", &ModulesIntegration{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Push events via API.
|
|
module.EventRestart.AddCallback("expose restart event", func(wc *mgr.WorkerCtx, s struct{}) (cancel bool, err error) {
|
|
// Send event as runtime:modules/core/event/restart
|
|
pushModuleEvent("core", "restart", false, nil)
|
|
return false, nil
|
|
})
|
|
module.EventShutdown.AddCallback("expose shutdown event", func(wc *mgr.WorkerCtx, s struct{}) (cancel bool, err error) {
|
|
// Send event as runtime:modules/core/event/shutdown
|
|
pushModuleEvent("core", "shutdown", false, nil)
|
|
return false, nil
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
// ModulesIntegration provides integration with the modules system.
|
|
type ModulesIntegration struct{}
|
|
|
|
// Set is called when the value is set from outside.
|
|
// If the runtime value is considered read-only ErrReadOnly
|
|
// should be returned. It is guaranteed that the key of
|
|
// the record passed to Set is prefixed with the key used
|
|
// to register the value provider.
|
|
func (mi *ModulesIntegration) Set(record.Record) (record.Record, error) {
|
|
return nil, runtime.ErrReadOnly
|
|
}
|
|
|
|
// Get should return one or more records that match keyOrPrefix.
|
|
// keyOrPrefix is guaranteed to be at least the prefix used to
|
|
// register the ValueProvider.
|
|
func (mi *ModulesIntegration) Get(keyOrPrefix string) ([]record.Record, error) {
|
|
return nil, database.ErrNotFound
|
|
}
|
|
|
|
type eventData struct {
|
|
record.Base
|
|
sync.Mutex
|
|
Data interface{}
|
|
}
|
|
|
|
func pushModuleEvent(moduleName, eventName string, internal bool, data interface{}) {
|
|
// Create event record and set key.
|
|
eventRecord := &eventData{
|
|
Data: data,
|
|
}
|
|
eventRecord.SetKey(fmt.Sprintf(
|
|
"runtime:modules/%s/event/%s",
|
|
moduleName,
|
|
eventName,
|
|
))
|
|
eventRecord.UpdateMeta()
|
|
if internal {
|
|
eventRecord.Meta().MakeSecret()
|
|
eventRecord.Meta().MakeCrownJewel()
|
|
}
|
|
|
|
// Push event to database subscriptions.
|
|
modulesIntegrationUpdatePusher(eventRecord)
|
|
}
|