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

Merge a52130f5de77731108d4ec58e44d3ae70f61989d into 537afbe95b2019ac8c32e700bd6d7dc2929e3187

Browse Source
This commit is contained in:
Francis Mendoza 2024-11-16 04:51:24 +00:00 committed by GitHub
commit 0e6f8797bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/wgsd-client/README.md
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
``` ```

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 {