github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 10:46:57 -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

17
modules/dhcp6_spoof.go
View file

@ -369,9 +369,7 @@ func (s *DHCP6Spoofer) Start() error {
return err
}
s.SetRunning(true)
go func() {
return s.SetRunning(true, func() {
defer s.Handle.Close()
src := gopacket.NewPacketSource(s.Handle, s.Handle.LinkType())
@ -382,16 +380,11 @@ func (s *DHCP6Spoofer) Start() error {
s.onPacket(packet)
}
}()
return nil
})
}
func (s *DHCP6Spoofer) Stop() error {
if s.Running() == false {
return session.ErrAlreadyStopped
}
s.SetRunning(false)
s.Handle.Close()
return nil
return s.SetRunning(false, func() {
s.Handle.Close()
})
}