github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-19 21:13:18 -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

21
modules/net_recon.go
View file

@ -125,15 +125,11 @@ func (d *Discovery) Configure() error {
}
func (d *Discovery) Start() error {
if d.Running() == true {
return session.ErrAlreadyStarted
} else if err := d.Configure(); err != nil {
if err := d.Configure(); err != nil {
return err
}
d.SetRunning(true)
go func() {
return d.SetRunning(true, func() {
for {
select {
case <-time.After(time.Duration(d.refresh) * time.Second):
@ -152,16 +148,11 @@ func (d *Discovery) Start() error {
return
}
}
}()
return nil
})
}
func (d *Discovery) Stop() error {
if d.Running() == false {
return session.ErrAlreadyStopped
}
d.quit <- true
d.SetRunning(false)
return nil
return d.SetRunning(false, func() {
d.quit <- true
})
}