github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-06 21:12:05 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: -silent and -debug are now mapped to log.silent and log.debug that can be set at runtime (fixes #50).

Browse source
This commit is contained in:
evilsocket 2018-02-15 20:48:34 +01:00
parent 09808be1a4
commit 21236c257e
3 changed files with 58 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
session/environment.go
View file

@ -9,12 +9,16 @@ import (
"github.com/evilsocket/bettercap-ng/core"
)
type SetCallback func(newValue string)
type Environment struct {
sync.Mutex
Padding int `json:"-"`
Storage map[string]string `json:"storage"`
sess *Session
cbs map[string]SetCallback
sess *Session
}
func NewEnvironment(s *Session) *Environment {
@ -22,6 +26,7 @@ func NewEnvironment(s *Session) *Environment {
Padding: 0,
Storage: make(map[string]string),
sess: s,
cbs: make(map[string]SetCallback),
}
return env
@ -36,6 +41,18 @@ func (env *Environment) Has(name string) bool {
return found
}
func (env *Environment) SetCallback(name string, cb SetCallback) {
env.Lock()
defer env.Unlock()
env.cbs[name] = cb
}
func (env *Environment) WithCallback(name, value string, cb SetCallback) string {
ret := env.Set(name, value)
env.SetCallback(name, cb)
return ret
}
func (env *Environment) Set(name, value string) string {
env.Lock()
defer env.Unlock()
@ -43,6 +60,10 @@ func (env *Environment) Set(name, value string) string {
old, _ := env.Storage[name]
env.Storage[name] = value
if cb, hasCallback := env.cbs[name]; hasCallback == true {
cb(value)
}
env.sess.Events.Log(core.DEBUG, "env.change: %s -> '%s'", name, value)
width := len(name)