astra/doggo
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

chore: Fix UDP Port parsing

Browse source
This commit is contained in:
Karan Sharma 2020-12-11 16:05:16 +05:30
parent 13cf2caaee
commit a57814688c
3 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cmd/parse.go
View file

@ -32,7 +32,7 @@ func (hub *Hub) loadQueryArgs(c *cli.Context) error {
func (hub *Hub) parseFreeArgs(c *cli.Context) error {
for _, arg := range c.Args().Slice() {
if strings.HasPrefix(arg, "@") {
hub.QueryFlags.Nameservers.Set(arg)
hub.QueryFlags.Nameservers.Set(strings.Trim(arg, "@"))
} else if _, ok := dns.StringToType[strings.ToUpper(arg)]; ok {
hub.QueryFlags.QTypes.Set(arg)
} else if _, ok := dns.StringToClass[strings.ToUpper(arg)]; ok {