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

refact: refactored module SetRunning method (fixes #49)

Browse source
This commit is contained in:
evilsocket 2018-02-11 01:43:48 +01:00
commit 9c6eb70eb3
15 changed files with 103 additions and 174 deletions
Download patch file Download diff file Expand all files Collapse all files

31
modules/arp_spoof.go
View file

@ -173,14 +173,11 @@ func (p *ArpSpoofer) Configure() error {
}
func (p *ArpSpoofer) Start() error {
if p.Running() == true {
return session.ErrAlreadyStarted
} else if err := p.Configure(); err != nil {
if err := p.Configure(); err != nil {
return err
}
p.SetRunning(true)
go func() {
return p.SetRunning(true, func() {
from := p.Session.Gateway.IP
from_hw := p.Session.Interface.HW
@ -192,24 +189,14 @@ func (p *ArpSpoofer) Start() error {
}
p.done <- true
}()
return nil
})
}
func (p *ArpSpoofer) Stop() error {
if p.Running() == false {
return session.ErrAlreadyStopped
}
log.Info("Waiting for ARP spoofer to stop ...")
p.SetRunning(false)
<-p.done
p.unSpoof()
p.ban = false
return nil
return p.SetRunning(false, func() {
log.Info("Waiting for ARP spoofer to stop ...")
<-p.done
p.unSpoof()
p.ban = false
})
}