vikarti.anatra/safing-portmaster
1
0
Fork 0
mirror of https://github.com/safing/portmaster synced 2025-09-15 17:29:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Always trigger event on shutdown

Browse source
This commit is contained in:
Daniel 2021-10-07 14:26:42 +02:00
parent 77d5b083eb
commit 4c9abbabd3
2 changed files with 34 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

31
core/api.go
View file

@ -1,9 +1,7 @@
package core
import (
"context"
"net/http"
"time"
"github.com/safing/portbase/api"
"github.com/safing/portbase/log"
@ -13,16 +11,6 @@ import (
"github.com/safing/portmaster/updates"
)
const (
eventShutdown = "shutdown"
eventRestart = "restart"
)
func registerEvents() {
module.RegisterEvent(eventShutdown, true)
module.RegisterEvent(eventRestart, true)
}
func registerAPIEndpoints() error {
if err := api.RegisterEndpoint(api.Endpoint{
Path: "core/shutdown",
@ -70,16 +58,8 @@ func registerAPIEndpoints() error {
func shutdown(_ *api.Request) (msg string, err error) {
log.Warning("core: user requested shutdown via action")
module.StartWorker("shutdown", func(context.Context) error {
// Notify everyone of the shutdown.
module.TriggerEvent(eventShutdown, nil)
// Wait a bit for the event to propagate.
time.Sleep(1 * time.Second)
// Do not run in worker, as this would block itself here.
go modules.Shutdown() //nolint:errcheck
return nil
})
return "shutdown initiated", nil
}
@ -88,15 +68,10 @@ func shutdown(_ *api.Request) (msg string, err error) {
func restart(_ *api.Request) (msg string, err error) {
log.Info("core: user requested restart via action")
module.StartWorker("restart", func(context.Context) error {
// Notify everyone of the shutdown.
module.TriggerEvent(eventRestart, nil)
// Wait a bit for the event to propagate.
time.Sleep(1 * time.Second)
// Trigger restart event instead of shutdown event.
restarting.Set()
// Let the updates module handle restarting.
updates.RestartNow()
return nil
})
return "restart initiated", nil
}

28
core/core.go
View file

@ -2,9 +2,11 @@ package core
import (
"fmt"
"time"
"github.com/safing/portbase/modules"
"github.com/safing/portbase/modules/subsystems"
"github.com/tevino/abool"
// module dependencies
_ "github.com/safing/portmaster/netenv"
@ -13,8 +15,15 @@ import (
_ "github.com/safing/portmaster/updates"
)
const (
eventShutdown = "shutdown"
eventRestart = "restart"
)
var (
module *modules.Module
restarting = abool.New()
)
func init() {
@ -27,6 +36,8 @@ func init() {
"config:core/",
nil,
)
modules.SetGlobalShutdownFn(shutdownHook)
}
func prep() error {
@ -54,3 +65,20 @@ func start() error {
return nil
}
func registerEvents() {
module.RegisterEvent(eventShutdown, true)
module.RegisterEvent(eventRestart, true)
}
func shutdownHook() {
// Notify everyone of the restart/shutdown.
if restarting.IsNotSet() {
module.TriggerEvent(eventShutdown, nil)
} else {
module.TriggerEvent(eventRestart, nil)
}
// Wait a bit for the event to propagate.
time.Sleep(1 * time.Second)
}