github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-21 05:53:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

misc: small fix or general refactoring i did not bother commenting

Browse source
This commit is contained in:
Simone Margaritelli 2024-09-21 11:56:14 +02:00
commit 2966153adf
8 changed files with 320 additions and 230 deletions
Download patch file Download diff file Expand all files Collapse all files

7
modules/zerogod/zerogod_save.go
View file

@ -1,6 +1,7 @@
package zerogod
import (
"errors"
"fmt"
"io/ioutil"
@ -32,6 +33,10 @@ func svcEntriesToData(services map[string]*zeroconf.ServiceEntry) []ServiceData
}
func (mod *ZeroGod) save(address, filename string) error {
if mod.browser == nil {
return errors.New("use 'zerogod.discovery on' to start the discovery first")
}
if address == "" {
return fmt.Errorf("address cannot be empty")
}
@ -39,7 +44,7 @@ func (mod *ZeroGod) save(address, filename string) error {
return fmt.Errorf("filename cannot be empty")
}
if ipServices, found := mod.mapping[address]; found {
if ipServices := mod.browser.GetServicesFor(address); ipServices != nil {
services := svcEntriesToData(ipServices)
data, err := yaml.Marshal(services)
if err != nil {