github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-20 13:33:21 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: new -env-file for env persistance (closes #81)

Browse source
This commit is contained in:
evilsocket 2018-02-25 02:15:22 +01:00
commit 5509a58e52
3 changed files with 50 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
session/session.go
View file

@ -142,7 +142,7 @@ func New() (*Session, error) {
}
}
s.Env = NewEnvironment(s)
s.Env = NewEnvironment(s, *s.Options.EnvFile)
s.Events = NewEventPool(*s.Options.Debug, *s.Options.Silent)
s.registerCoreHandlers()
@ -246,6 +246,13 @@ func (s *Session) Close() {
s.Firewall.Restore()
s.Queue.Stop()
if *s.Options.EnvFile != "" {
envFile, _ := core.ExpandPath(*s.Options.EnvFile)
if err := s.Env.Save(envFile); err != nil {
fmt.Printf("Error while storing the environment to %s: %s", envFile, err)
}
}
if *s.Options.CpuProfile != "" {
pprof.StopCPUProfile()
}
@ -304,7 +311,6 @@ func (s *Session) setupSignals() {
}
func (s *Session) setupEnv() {
s.Env.Set(PromptVariable, DefaultPrompt)
s.Env.Set("iface.index", fmt.Sprintf("%d", s.Interface.Index))
s.Env.Set("iface.name", s.Interface.Name())
s.Env.Set("iface.ipv4", s.Interface.IpAddress)
@ -313,6 +319,10 @@ func (s *Session) setupEnv() {
s.Env.Set("gateway.address", s.Gateway.IpAddress)
s.Env.Set("gateway.mac", s.Gateway.HwAddress)
if found, v := s.Env.Get(PromptVariable); found == false || v == "" {
s.Env.Set(PromptVariable, DefaultPrompt)
}
dbg := "false"
if *s.Options.Debug {
dbg = "true"