fix: support ipv6 on argument line as well.

ipinfo · Aug 16, 2021 · dc98712 · dc98712
1 parent 91eeb9a
commit dc98712
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/lib/cmd_cidr2range.go b/lib/cmd_cidr2range.go
@@ -114,9 +114,16 @@ func CmdCIDR2Range(
 		f, err := os.Open(arg)
 		if err != nil {
 			// is it a CIDR?
-			if r, err := IPRangeStrFromCIDR(arg); err == nil {
-				fmt.Println(r.String())
-				continue
+			if strings.IndexByte(arg, ':') == -1 {
+				if r, err := IPRangeStrFromCIDR(arg); err == nil {
+					fmt.Println(r.String())
+					continue
+				}
+			} else {
+				if r, err := IP6RangeStrFromCIDR(arg); err == nil {
+					fmt.Println(r.String())
+					continue
+				}
 			}
 
 			// invalid file arg.