Bugfix: User can't change DNS Invaild Settings (#1042)
* Fix core restart panic on invalid DNS config * Avoid nil instance race after core stop
This commit is contained in:
+16
-1
@@ -2,6 +2,7 @@ package core
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"sync"
|
||||||
|
|
||||||
"github.com/alireza0/s-ui/logger"
|
"github.com/alireza0/s-ui/logger"
|
||||||
|
|
||||||
@@ -28,6 +29,7 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Core struct {
|
type Core struct {
|
||||||
|
mu sync.RWMutex
|
||||||
isRunning bool
|
isRunning bool
|
||||||
instance *Box
|
instance *Box
|
||||||
}
|
}
|
||||||
@@ -46,10 +48,15 @@ func (c *Core) GetCtx() context.Context {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Core) GetInstance() *Box {
|
func (c *Core) GetInstance() *Box {
|
||||||
|
c.mu.RLock()
|
||||||
|
defer c.mu.RUnlock()
|
||||||
return c.instance
|
return c.instance
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Core) Start(sbConfig []byte) error {
|
func (c *Core) Start(sbConfig []byte) error {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
|
||||||
var opt option.Options
|
var opt option.Options
|
||||||
err := opt.UnmarshalJSONContext(globalCtx, sbConfig)
|
err := opt.UnmarshalJSONContext(globalCtx, sbConfig)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -66,6 +73,7 @@ func (c *Core) Start(sbConfig []byte) error {
|
|||||||
|
|
||||||
err = c.instance.Start()
|
err = c.instance.Start()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
c.instance = nil
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -81,13 +89,20 @@ func (c *Core) Start(sbConfig []byte) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Core) Stop() error {
|
func (c *Core) Stop() error {
|
||||||
|
c.mu.Lock()
|
||||||
|
defer c.mu.Unlock()
|
||||||
|
|
||||||
c.isRunning = false
|
c.isRunning = false
|
||||||
if c.instance == nil {
|
if c.instance == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return c.instance.Close()
|
err := c.instance.Close()
|
||||||
|
c.instance = nil
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Core) IsRunning() bool {
|
func (c *Core) IsRunning() bool {
|
||||||
|
c.mu.RLock()
|
||||||
|
defer c.mu.RUnlock()
|
||||||
return c.isRunning
|
return c.isRunning
|
||||||
}
|
}
|
||||||
|
|||||||
+1
-1
@@ -117,7 +117,7 @@ func (s *ConfigService) restartCoreWithConfig(config json.RawMessage) error {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
if corePtr.IsRunning() {
|
if corePtr.IsRunning() {
|
||||||
err = corePtr.GetInstance().Close()
|
err = corePtr.Stop()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user