This commit is contained in:
Francis Mendoza
2024-11-16 04:51:24 +00:00
committed by GitHub
2 changed files with 5 additions and 0 deletions

View File

@@ -10,4 +10,6 @@ Usage of ./wgsd-client:
ip:port of DNS server ip:port of DNS server
-zone string -zone string
dns zone name dns zone name
-protocol string (optional)
udp (default) or tcp
``` ```

View File

@@ -24,6 +24,8 @@ var (
dnsServerFlag = flag.String("dns", "", dnsServerFlag = flag.String("dns", "",
"ip:port of DNS server") "ip:port of DNS server")
dnsZoneFlag = flag.String("zone", "", "dns zone name") dnsZoneFlag = flag.String("zone", "", "dns zone name")
// Adding optional protocol flag to specify protocol to use
protocolFlag = flag.String("protocol", "", "optional: tcp or udp (default)")
) )
func main() { func main() {
@@ -63,6 +65,7 @@ func main() {
defer close(done) defer close(done)
dnsClient := &dns.Client{ dnsClient := &dns.Client{
Timeout: time.Second * 5, Timeout: time.Second * 5,
Net: *protocolFlag, // Inserting string value of protocol flag if present to indicate udp, tcp, etc.
} }
for _, peer := range wgDevice.Peers { for _, peer := range wgDevice.Peers {
select { select {