github/jwhited-wgsd
1
0
Fork 0
mirror of https://github.com/jwhited/wgsd.git synced 2025-04-04 11:09:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added ability to query over tcp via optional protocol flag

Browse Source
This commit is contained in:
Francis Mendoza 2024-11-15 20:29:56 -08:00
parent 537afbe95b
commit 8ed190f81a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cmd/wgsd-client/main.go
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 {