fix: gracefully handling hid receiver disconnection

This commit is contained in:
evilsocket 2019-03-30 16:27:56 +01:00
parent afe300cd8a
commit 0113286b4f
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
2 changed files with 23 additions and 1 deletions

View file

@ -204,11 +204,21 @@ func (mod *HIDRecon) Configure() error {
return nil
}
func (mod *HIDRecon) forceStop() error {
return mod.SetRunning(false, func() {
if mod.dongle != nil {
mod.dongle.Close()
mod.dongle = nil
mod.Debug("device closed")
}
})
}
func (mod *HIDRecon) Stop() error {
return mod.SetRunning(false, func() {
mod.waitGroup.Wait()
if mod.dongle != nil {
mod.dongle.Close()
mod.dongle = nil
mod.Debug("device closed")
}
})