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

21
modules/mac_changer.go
View file

@ -103,28 +103,23 @@ func (mc *MacChanger) setMac(mac net.HardwareAddr) error {
}
func (mc *MacChanger) Start() error {
if mc.Running() == true {
return session.ErrAlreadyStarted
} else if err := mc.Configure(); err != nil {
if err := mc.Configure(); err != nil {
return err
} else if err := mc.setMac(mc.fakeMac); err != nil {
return err
}
mc.SetRunning(true)
log.Info("Interface mac address set to %s", core.Bold(mc.fakeMac.String()))
return nil
return mc.SetRunning(true, func() {
log.Info("Interface mac address set to %s", core.Bold(mc.fakeMac.String()))
})
}
func (mc *MacChanger) Stop() error {
if mc.Running() == false {
return session.ErrAlreadyStopped
} else if err := mc.setMac(mc.originalMac); err != nil {
if err := mc.setMac(mc.originalMac); err != nil {
return err
}
mc.SetRunning(false)
log.Info("Interface mac address restored to %s", core.Bold(mc.originalMac.String()))
return nil
return mc.SetRunning(false, func() {
log.Info("Interface mac address restored to %s", core.Bold(mc.originalMac.String()))
})
}