trivial fixes

This commit is contained in:
Jordan Whited
2020-05-15 13:01:58 -07:00
parent 53c1794798
commit 5c7c582a2a
2 changed files with 7 additions and 2 deletions

View File

@@ -26,11 +26,12 @@ var (
)
func main() {
flag.Parse()
if len(*deviceFlag) < 1 {
log.Fatal("missing device flag")
}
if len(*dnsZoneFlag) < 1 {
log.Fatal("missing dns flag")
log.Fatal("missing zone flag")
}
resolver := net.DefaultResolver
if len(*dnsServerFlag) > 0 {
@@ -59,6 +60,10 @@ func main() {
"error retrieving Wireguard device '%s' info: %v",
*deviceFlag, err)
}
if len(wgDevice.Peers) < 1 {
log.Println("no peers found")
os.Exit(0)
}
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
done := make(chan struct{})